another typo

This commit is contained in:
Dr. Kashif Rasul
2020-01-16 10:51:16 +01:00
parent 177834451a
commit e34561373e
+1 -1
View File
@@ -111,7 +111,7 @@ class PTSEstimator(Estimator):
training_iter_dataset,
batch_size=self.trainer.batch_size,
num_workers=self.trainer.num_workers,
pip_memory=self.trainer.pip_memory
pin_memory=self.trainer.pin_memory
)
# ensure that the training network is created on the same device