make random crop in augmentation a bit less severe

lucidrains · Aug 23, 2020 · 162d592 · 162d592
1 parent 40168eb
commit 162d592
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -4,7 +4,7 @@
   name = 'stylegan2_pytorch',
   packages = find_packages(),
   scripts=['bin/stylegan2_pytorch'],
-  version = '0.20.0',
+  version = '0.20.1',
   license='GPLv3+',
   description = 'StyleGan2 in Pytorch',
   author = 'Phil Wang',

diff --git a/stylegan2_pytorch/stylegan2_pytorch.py b/stylegan2_pytorch/stylegan2_pytorch.py
@@ -288,7 +288,7 @@ def __init__(self, D, image_size):
 
     def forward(self, images, prob = 0., detach = False):
         if random() < prob:
-            random_scale = random_float(0.5, 0.9)
+            random_scale = random_float(0.75, 0.95)
             images = random_hflip(images, prob=0.5)
             images = random_crop_and_resize(images, scale = random_scale)