updated result, requirements and torch version

2023-03-24 07:51:57 +09:00
parent 2504426164
commit 942eae10a9
6 changed files with 36 additions and 25 deletions
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,11 +1,11 @@
-torch==1.13.0
+torch==2.0.0
+torchvision==0.15.1
 numpy==1.20.1
-torchvision==0.14.0
 tensorboard==2.5.0
 pandas==1.2.4
 matplotlib==3.4.1
 ruamel.yaml==0.17.4
-gym[atari]==0.18.0
+gym[atari]==0.17.0
 moviepy==1.0.3
 einops==0.3.0
 protobuf==3.20.0