once more, with eeling

ajbrock · Apr 5, 2019 · f6f9198 · f6f9198
1 parent 8c940f4
commit f6f9198
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/utils.py b/utils.py
@@ -886,7 +886,7 @@ def sample_sheet(G, classes_per_sheet, num_classes, samples_per_class, parallel,
     ims = []
     y = torch.arange(i * classes_per_sheet, (i + 1) * classes_per_sheet, device='cuda')
     for j in range(samples_per_class):
-      if (z is not None) and hasattr(z, 'sample_') and classes_per_sheet > z_.size(0):
+      if (z_ is not None) and hasattr(z_, 'sample_') and classes_per_sheet > z_.size(0):
         z_.sample_()
       else:
         z_ = torch.randn(classes_per_sheet, G.dim_z, device='cuda')