attempt to fix issue with torch 1.7, ddp and detach_()

lucidrains · lucidrains · commit ae7918f2568f · 2020-11-09T21:33:37.000-08:00
diff --git a/stylegan2_pytorch/stylegan2_pytorch.py b/stylegan2_pytorch/stylegan2_pytorch.py
@@ -330,7 +330,7 @@ def forward(self, images, prob = 0., types = [], detach = False):
             images = DiffAugment(images, types=types)
 
         if detach:
-            images.detach_()
+            images = images.detach()
 
         return self.D(images)
 
diff --git a/stylegan2_pytorch/version.py b/stylegan2_pytorch/version.py
@@ -1 +1 @@
-__version__ = '1.2.7'
+__version__ = '1.2.8'
