Merge pull request #10 from kashif/patch-1

ikostrikov2 · web-flow · commit 34a34a0f25a8 · 2019-12-24T14:32:42.000-05:00
alternate the mask for each stack in glow
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()

Original file line number	Diff line number	Diff line change
`@@ -155,7 +155,7 @@`
`155`	`155`	`num_inputs, num_hidden, mask, num_cond_inputs,`
`156`	`156`	`s_act='tanh', t_act='relu')`
`157`	`157`	`]`
`158`		`- mask = 1 - mask`
	`158`	`+ mask = 1 - mask`
`159`	`159`	`elif args.flow == 'realnvp':`
`160`	`160`	`mask = torch.arange(0, num_inputs) % 2`
`161`	`161`	`mask = mask.to(device).float()`