From 50683efdc699ea49ea79be3055365b7ec111e2c5 Mon Sep 17 00:00:00 2001 From: Edrin Basha Date: Tue, 27 Oct 2020 12:37:11 +0100 Subject: [PATCH] Update DeepARPredictionNetwork (#25) Set default value of num_parallel_samples to 100 --- pts/model/deepvar/deepvar_network.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pts/model/deepvar/deepvar_network.py b/pts/model/deepvar/deepvar_network.py index d37b0e5..1f403d5 100644 --- a/pts/model/deepvar/deepvar_network.py +++ b/pts/model/deepvar/deepvar_network.py @@ -422,7 +422,7 @@ class DeepVARTrainingNetwork(nn.Module): class DeepVARPredictionNetwork(DeepVARTrainingNetwork): - def __init__(self, num_parallel_samples: int, **kwargs) -> None: + def __init__(self, num_parallel_samples: int = 100, **kwargs) -> None: super().__init__(**kwargs) self.num_parallel_samples = num_parallel_samples