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
parent ab1238393c
commit b662f0e625
10 changed files with 753 additions and 196 deletions
--- a/setup.cfg
+++ b/setup.cfg
@@ -15,6 +15,7 @@ known_third_party =
    packaging
    PIL
    psutil
+    pytorch_lightning
    seqeval
    sklearn
    tensorboardX
@@ -23,6 +24,7 @@ known_third_party =
    torch
    torchtext
    torchvision
+    torch_xla

 line_length = 119
 lines_after_imports = 2