From 0d183bb5ea2fbe069e1c6806c4a9a1fd8e81656f Mon Sep 17 00:00:00 2001 From: Hyeonwoo Kang Date: Tue, 12 Jun 2018 18:57:00 +0900 Subject: [PATCH] Update to Pytorch 0.4 --- ACGAN.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/ACGAN.py b/ACGAN.py index e7a2eb2..09168be 100644 --- a/ACGAN.py +++ b/ACGAN.py @@ -161,9 +161,8 @@ def train(self): break z_ = torch.rand((self.batch_size, self.z_dim)) y_vec_ = torch.zeros((self.batch_size, self.class_num)).scatter_(1, y_.type(torch.LongTensor).unsqueeze(1), 1) - y_fill_ = y_vec_.unsqueeze(2).unsqueeze(3).expand(self.batch_size, self.class_num, self.input_size, self.input_size) if self.gpu_mode: - x_, z_, y_vec_, y_fill_ = x_.cuda(), z_.cuda(), y_vec_.cuda(), y_fill_.cuda() + x_, z_, y_vec_ = x_.cuda(), z_.cuda(), y_vec_.cuda() # update D network self.D_optimizer.zero_grad() @@ -261,4 +260,4 @@ def load(self): save_dir = os.path.join(self.save_dir, self.dataset, self.model_name) self.G.load_state_dict(torch.load(os.path.join(save_dir, self.model_name + '_G.pkl'))) - self.D.load_state_dict(torch.load(os.path.join(save_dir, self.model_name + '_D.pkl'))) \ No newline at end of file + self.D.load_state_dict(torch.load(os.path.join(save_dir, self.model_name + '_D.pkl')))