From d95a21780138b019b2ae25cb1e1c7aa3edc45062 Mon Sep 17 00:00:00 2001 From: "Dr. Kashif Rasul" Date: Wed, 29 Jan 2020 15:04:07 +0100 Subject: [PATCH] another typo --- pts/model/transformer_tempflow/transformer_tempflow_network.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pts/model/transformer_tempflow/transformer_tempflow_network.py b/pts/model/transformer_tempflow/transformer_tempflow_network.py index cd82f10..7c8ac5a 100644 --- a/pts/model/transformer_tempflow/transformer_tempflow_network.py +++ b/pts/model/transformer_tempflow/transformer_tempflow_network.py @@ -583,7 +583,7 @@ class TransformerTempFlowPredictionNetwork(TransformerTempFlowTrainingNetwork): # unroll_length=1, # ) - distr_args = self.distr_args(dec_output=dec_output) + distr_args = self.distr_args(decoder_output=dec_output) # (batch_size, 1, target_dim) new_samples = self.flow.sample(cond=distr_args)