fixed parameter deprecation in training/generation script of SpatialA…

…ttentionGAN (#7) * Update generate.py fix the deprecated parameters to make it compatible with the current version * Update train.py fix deprecated parameters * Update requirements.txt * Update train.py * Update generate.py
elvisyjlin · Jan 3, 2024 · 80cba20 · 80cba20
1 parent 80d453b
commit 80cba20
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/generate.py b/generate.py
@@ -93,4 +93,4 @@ def parse():
                     padding=0,
                     normalize=True,
                     range=(-1.,1.)
-                )
+                )
diff --git a/requirements.txt b/requirements.txt
@@ -1,7 +1,7 @@
 numpy>=1.14.0
 scikit-image>=0.14.2
 tqdm
-torch
-torchvision
+torch==1.11.0
+torchvision==0.12.0
 torchsummary
 tensorboardX