ignitenlp

Archived This repository is archived: read-only and no longer maintained.

Overview

In this example, we show how to use Ignite to finetune a transformer model:

Alternatively, install the all requirements using pip install -r requirements.txt.

Run the example on a single GPU:

If needed, please, adjust the batch size to your GPU device with --batch_size argument.

The default model is bert-base-uncased incase you need to change that use the --model argument, for details on which models can be used refer here

For details on accepted arguments:

Let's start training on a single node with 2 gpus:

Please, make sure to have Horovod installed before running.

Let's start training on a single node with 2 gpus:

For more details, see https://allegro.ai/clearml/docs/docs/examples/reporting/artifacts.html

From the project’s README on GitHub.

At a glance

RepositoryKentucky-Open-Science/ignitenlp
Research areaLanguage models & AI tooling
Primary languagePython
LicenseNone specified
Stars / forks0 / 0
Open issues and pull requests0
Created2022-04-04
Last push2022-04-05
Default branchmaster
Topicsnlp

What the README covers

Top contributors

Get the code

git clone https://github.com/Kentucky-Open-Science/ignitenlp.git