Merge pull request #10 from kashif/patch-1

alternate the mask for each stack in glow
ikostrikov · Dec 24, 2019 · 34a34a0 · 34a34a0
2 parents 6966665 + ff62f6b
commit 34a34a0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/main.py b/main.py
@@ -155,7 +155,7 @@
                 num_inputs, num_hidden, mask, num_cond_inputs,
                 s_act='tanh', t_act='relu')
         ]
-    mask = 1 - mask
+        mask = 1 - mask
 elif args.flow == 'realnvp':
     mask = torch.arange(0, num_inputs) % 2
     mask = mask.to(device).float()