This commit is contained in:
Dr. Kashif Rasul
2020-01-16 10:41:38 +01:00
parent 5eacc141ac
commit 28f6ed1dbc
2 changed files with 2 additions and 1 deletions
+1
View File
@@ -204,3 +204,4 @@ class MultivariateGrouper:
freq=data[FieldName.START].freq,
),
)
+1 -1
View File
@@ -108,7 +108,7 @@ class PTSEstimator(Estimator):
)
training_data_loader = DataLoader(
traning_iter_dataset,
training_iter_dataset,
batch_size=self.trainer.batch_size,
num_workers=self.train.num_workers,
pip_memory=self.trainer.pip_memory