diff --git a/scripts/pytorch_lightning_training_script/requirements.txt b/scripts/pytorch_lightning_training_script/requirements.txt index 9bc733e..8dc126a 100644 --- a/scripts/pytorch_lightning_training_script/requirements.txt +++ b/scripts/pytorch_lightning_training_script/requirements.txt @@ -1,6 +1,6 @@ torch ==1.6.0 transformers == 3.4.0 -pytorch-lightning == 1.0.3 +pytorch-lightning == 1.6.0 # allennlp 0.9 with gradient accumulation and fp16 # need