diff --git a/stargan/solver.py b/stargan/solver.py index b0fdae6..0dd09b4 100644 --- a/stargan/solver.py +++ b/stargan/solver.py @@ -594,7 +594,7 @@ class Solver(object): for i, (x_real, c_org) in enumerate(data_loader): # Black image - black = np.ones((1,3,256,256)) + black = np.zeros((1,3,256,256)) black = torch.FloatTensor(black).to(self.device) # Prepare input images and target domain labels.