diff --git a/requirements.txt b/requirements.txt index 54a48fd..d498d63 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,5 +1,5 @@ torchvision==0.6.0 torchsummary==1.5.1 pytorch_lightning==0.7.5 -torch==1.5.0 +torch==2.2.0 tensorboard==2.2.1 \ No newline at end of file