From 054104019ca5bc4b73fa32cf5476b0667a007d49 Mon Sep 17 00:00:00 2001 From: "Dr. Kashif Rasul" Date: Wed, 29 Jan 2020 14:47:07 +0100 Subject: [PATCH] fix typo --- pts/model/tempflow/tempflow_network.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pts/model/tempflow/tempflow_network.py b/pts/model/tempflow/tempflow_network.py index 72f769d..4a4fc31 100644 --- a/pts/model/tempflow/tempflow_network.py +++ b/pts/model/tempflow/tempflow_network.py @@ -65,7 +65,7 @@ class TempFlowTrainingNetwork(nn.Module): self.dequantize = dequantize self.distr_output = FlowOutput( - self.flow, input_size=input_size, cond_size=conditioning_length + self.flow, input_size=target_dim, cond_size=conditioning_length ) self.proj_dist_args = self.distr_output.get_args_proj(num_cells)