update

2022-01-10 17:34:43 +08:00
parent 591c650dd9
commit bf2df5c5a6
3 changed files with 212 additions and 3 deletions
@@ -256,7 +256,7 @@ class Trainer(object):

                latent_id       = F.normalize(latent_id, p=2, dim=1)
        
-                losses, img_fake= model(None, src_image1, latent_id, None, for_G=True)
+                losses, img_fake= self.gen(src_image1, latent_id)

                # update Generator weights
                losses      = [ torch.mean(x) if not isinstance(x, int) else x for x in losses ]