Fixed generate samples script

pdeubel · pdeubel · commit 5d9e780578ff · 2022-04-24T14:34:24.000+02:00
diff --git a/evaluation/visualization/generate_samples.py b/evaluation/visualization/generate_samples.py
@@ -41,7 +41,7 @@ def main(vae_dirs: List[str], gpu: int, best_vae: bool, seed: int):
         with torch.no_grad():
             sample = vae.decode(random_latent_vector)
 
-        _save_sample(sample, vae_version)
+        _save_sample(vae.denormalize(sample), vae_version)
 
 
 if __name__ == "__main__":
