fix for horizontal flip probability, thanks to @xiankgx

lucidrains · Jul 19, 2022 · 0588213 · 0588213
1 parent e7cbe03
commit 0588213
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -15,6 +15,7 @@
   version = __version__,
   license='GPLv3+',
   description = 'StyleGan2 in Pytorch',
+  long_description_content_type = 'text/markdown',
   author = 'Phil Wang',
   author_email = 'lucidrains@gmail.com',
   url = 'https://github.com/lucidrains/stylegan2-pytorch',

diff --git a/stylegan2_pytorch/stylegan2_pytorch.py b/stylegan2_pytorch/stylegan2_pytorch.py
@@ -387,7 +387,7 @@ def __getitem__(self, index):
 # augmentations
 
 def random_hflip(tensor, prob):
-    if prob > random():
+    if prob < random():
         return tensor
     return torch.flip(tensor, dims=(3,))
 

diff --git a/stylegan2_pytorch/version.py b/stylegan2_pytorch/version.py
@@ -1 +1 @@
-__version__ = '1.8.8'
+__version__ = '1.8.9'