Skip to content

Commit 162d592

Browse files
committed
make random crop in augmentation a bit less severe
1 parent 40168eb commit 162d592

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
name = 'stylegan2_pytorch',
55
packages = find_packages(),
66
scripts=['bin/stylegan2_pytorch'],
7-
version = '0.20.0',
7+
version = '0.20.1',
88
license='GPLv3+',
99
description = 'StyleGan2 in Pytorch',
1010
author = 'Phil Wang',

stylegan2_pytorch/stylegan2_pytorch.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -288,7 +288,7 @@ def __init__(self, D, image_size):
288288

289289
def forward(self, images, prob = 0., detach = False):
290290
if random() < prob:
291-
random_scale = random_float(0.5, 0.9)
291+
random_scale = random_float(0.75, 0.95)
292292
images = random_hflip(images, prob=0.5)
293293
images = random_crop_and_resize(images, scale = random_scale)
294294

0 commit comments

Comments
 (0)