bug fix

GlassyWing · Jan 18, 2021 · 0dfd39f · 0dfd39f
1 parent 960e72a
commit 0dfd39f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/random_sample.py b/random_sample.py
@@ -11,12 +11,12 @@
     model.apply(add_sn)
     model.to(device)
 
-    model.load_state_dict(torch.load("checkpoints/ae_ckpt_103_0.074130.pth", map_location=device), strict=False)
+    model.load_state_dict(torch.load("checkpoints/ae_ckpt_169_0.689621.pth", map_location=device), strict=False)
 
     model.eval()
 
     with torch.no_grad():
-        z = torch.randn((25, 512)).to(device)
+        z = torch.randn((25, 512, 2, 2)).to(device)
         gen_imgs, _ = model.decoder(z)
         gen_imgs = gen_imgs.permute(0, 2, 3, 1)
         for gen_img in gen_imgs: