Update utils.py

utils.py

@@ -13,4 +13,5 @@ def generate(gan, batch_size=1):
     with torch.no_grad():
         ims = gan.G(torch.randn(batch_size, gan.latent_dim)).clamp_(0.0, 1.0) * 255
         ims = ims.permute(0, 2, 3, 1).detach().cpu().numpy().astype(np.uint8)
-    return ims
+    return ims
+