diff --git a/requirements.txt b/requirements.txt index 0d561b0..523d421 100644 --- a/requirements.txt +++ b/requirements.txt @@ -4,5 +4,5 @@ tqdm==4.26.0 allennlp==0.8.5 pytorch-ignite==0.2.0 pytorch_transformers==1.1.0 -torch==1.1.0 +torch==2.2.0 PyYAML>=5.4