Fix Norm error

Lornatang · Jan 20, 2021 · 5301c3f · 5301c3f
1 parent a817276
commit 5301c3f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test.py b/test.py
@@ -63,7 +63,7 @@
     with torch.no_grad():
         generated_images = model(noise)
 
-    vutils.save_image(generated_images, f"{args.outf}/test.png")
+    vutils.save_image(generated_images, f"{args.outf}/test.png", normalize=True)
     print("##################################################\n")
 
     logger.info("Test completed successfully.\n")