wassname/pytorch-ts
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-ts.git synced 2026-07-16 11:21:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update DeepARPredictionNetwork (#25)

Browse Source 
Set default value of num_parallel_samples to 100
This commit is contained in:
Edrin Basha
2020-10-27 12:37:11 +01:00
committed by GitHub Enterprise
parent 2b957e89e3
commit 50683efdc6
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pts/model/deepvar/deepvar_network.py
+1 -1
View File
@@ -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