flow's input size is target_dim

2026-07-10 01:55:45 +08:00 · 2020-01-13 12:06:22 +01:00
parent 863c675981
commit 2f019604ff
1 changed files with 1 additions and 1 deletions
@@ -54,7 +54,7 @@ class TempFlowTrainingNetwork(nn.Module):

        flow_cls = {"RealNVP": RealNVP}[flow_type]
        self.flow = flow_cls(
-            input_size=input_size,
+            input_size=target_dim,
            n_blocks=n_blocks,
            n_hidden=n_hidden,
            hidden_size=hidden_size,