diff --git a/pts/model/transformer/transformer_network.py b/pts/model/transformer/transformer_network.py index bb44462..8de9f1e 100644 --- a/pts/model/transformer/transformer_network.py +++ b/pts/model/transformer/transformer_network.py @@ -418,7 +418,7 @@ class TransformerPredictionNetwork(TransformerNetwork): self, feat_static_cat: torch.Tensor, feat_static_real: torch.Tensor, - feature_static_real: torch.Tensor, + feat_static_real: torch.Tensor, past_time_feat: torch.Tensor, past_target: torch.Tensor, past_observed_values: torch.Tensor, @@ -441,7 +441,7 @@ class TransformerPredictionNetwork(TransformerNetwork): # create the inputs for the encoder inputs, scale, static_feat = self.create_network_input( feat_static_cat=feat_static_cat, - feature_static_real=feat_static_real, + feat_static_cat=feat_static_real, past_time_feat=past_time_feat, past_target=past_target, past_observed_values=past_observed_values, diff --git a/runs/Jan28_11-21-38_ZALANDO-61797/events.out.tfevents.1580206898.ZALANDO-61797.29012.0 b/runs/Jan28_11-21-38_ZALANDO-61797/events.out.tfevents.1580206898.ZALANDO-61797.29012.0 new file mode 100644 index 0000000..94e30d7 Binary files /dev/null and b/runs/Jan28_11-21-38_ZALANDO-61797/events.out.tfevents.1580206898.ZALANDO-61797.29012.0 differ