From 42f5dbd9547f124483fced71e6e09e43703ca267 Mon Sep 17 00:00:00 2001 From: Nataniel Ruiz Date: Tue, 24 Dec 2019 19:48:10 -0400 Subject: [PATCH] next --- stargan/solver.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/stargan/solver.py b/stargan/solver.py index 0dd09b4..4739762 100644 --- a/stargan/solver.py +++ b/stargan/solver.py @@ -597,6 +597,8 @@ class Solver(object): black = np.zeros((1,3,256,256)) black = torch.FloatTensor(black).to(self.device) + # black = torch.FloatTensor(torch.rand((1,3,256,256))).to(self.device) + # Prepare input images and target domain labels. x_real = x_real.to(self.device) c_trg_list = self.create_labels(c_org, self.c_dim, self.dataset, self.selected_attrs)