Update train_lapgan.py

witnessai · Mar 27, 2018 · 7f223f9 · 7f223f9
1 parent f0bfe10
commit 7f223f9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cifar10/alternative_loss_calculate-backward/train_lapgan.py b/cifar10/alternative_loss_calculate-backward/train_lapgan.py
@@ -214,7 +214,7 @@ def run_LAPGAN(n_level=3, n_epoch=1, batch_size=256, use_gpu=True,
 
     samples = LapGan_model.generate(n_sample)
     current_time = time.strftime('%Y-%m-%d %H%M%S')
-    samples = torch.Tensor(samples*3)
+    samples = torch.Tensor(samples)
     save_image(samples, './result/%s epoch%d.png'% (current_time, n_epoch), normalize=True)
     return samples.numpy()