updated results

genhao3 · Feb 10, 2020 · 55e654f · 55e654f
1 parent 7bb883a
commit 55e654f
Show file tree

Hide file tree

Showing 6 changed files with 6 additions and 6 deletions.
diff --git a/assets/LogCoshVAE_49.png b/assets/LogCoshVAE_49.png
diff --git a/assets/SWAE_49.png b/assets/SWAE_49.png
diff --git a/assets/recons_LogCoshVAE_49.png b/assets/recons_LogCoshVAE_49.png
diff --git a/assets/recons_SWAE_49.png b/assets/recons_SWAE_49.png
diff --git a/configs/swae.yaml b/configs/swae.yaml
@@ -4,7 +4,7 @@ model_params:
   latent_dim: 128
   reg_weight: 100
   wasserstein_deg: 2.0
-  num_projections: 100
+  num_projections: 200
   projection_dist: "normal" #"cauchy"
 
 exp_params:

diff --git a/experiment.py b/experiment.py
@@ -74,11 +74,11 @@ def sample_images(self):
                           normalize=True,
                           nrow=int(math.sqrt(self.params['batch_size'])))
 
-        vutils.save_image(test_input.data,
-                          f"{self.logger.save_dir}{self.logger.name}/version_{self.logger.version}/"
-                          f"real_img_{self.logger.name}_{self.current_epoch}.png",
-                          normalize=True,
-                          nrow=int(math.sqrt(self.params['batch_size'])))
+        # vutils.save_image(test_input.data,
+        #                   f"{self.logger.save_dir}{self.logger.name}/version_{self.logger.version}/"
+        #                   f"real_img_{self.logger.name}_{self.current_epoch}.png",
+        #                   normalize=True,
+        #                   nrow=int(math.sqrt(self.params['batch_size'])))
 
         samples = self.model.sample(self.params['batch_size'],
                                     self.curr_device,