fix for horizontal flip probability, thanks to @xiankgx

Author: lucidrains

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,))
 

stylegan2_pytorch/version.py

@@ -1 +1 @@
-__version__ = '1.8.8'
+__version__ = '1.8.9'