J.P Lee
2b60a26b46
Update examples/ner/run_ner.py to use AutoModel ( #3305 )
...
* Update examples/ner/run_ner.py to use AutoModel
* Fix missing code and apply `make style` command
2020-03-17 12:30:10 -04:00
Victor SANH
6b1ff25084
fix n_gpu count when no_cuda flag is activated ( #3077 )
...
* fix n_gpu count when no_cuda flag is activated
* someone was left behind
2020-03-02 10:20:21 -05:00
Martin Malmsten
d762d4289c
Code now passes style enforcement
2020-02-26 23:50:40 +01:00
Martin Malmsten
9495d38b0d
Changes from reviews.
2020-02-26 23:36:39 +01:00
Martin Malmsten
105dcb4162
Now passes style guide enforcement
2020-02-23 21:47:59 +01:00
Martin Malmsten
33eb8a165d
Added ,
2020-02-23 21:43:31 +01:00
Martin Malmsten
869b66f6b3
* Added support for Albert when fine-tuning for NER
...
* Added support for Albert in NER pipeline
* Added command-line options to examples/ner/run_ner.py to better control tokenization
* Added class AlbertForTokenClassification
* Changed output for NerPipeline to use .convert_ids_to_tokens(...) instead of .decode(...) to better reflect tokens
2020-02-23 21:13:03 +01:00
Martin Malmsten
4452b44b90
Labels are now added to model config under id2label and label2id ( #2945 )
2020-02-21 08:53:05 -05:00
srush
b662f0e625
Support for torch-lightning in NER examples ( #2890 )
...
* initial pytorch lightning commit
* tested multigpu
* Fix learning rate schedule
* black formatting
* fix flake8
* isort
* isort
* .
Co-authored-by: Check your git settings! <chris@chris-laptop>
2020-02-20 11:50:05 -05:00
