This commit is contained in:
Nataniel Ruiz
2019-12-24 19:34:26 -04:00
parent 6435aba0a9
commit da986b8641
+1 -1
View File
@@ -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.