From e34561373e9e47320e6e4768d8ea8a64671db1de Mon Sep 17 00:00:00 2001 From: "Dr. Kashif Rasul" Date: Thu, 16 Jan 2020 10:51:16 +0100 Subject: [PATCH] another typo --- pts/model/estimator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pts/model/estimator.py b/pts/model/estimator.py index 52720b6..07b047b 100644 --- a/pts/model/estimator.py +++ b/pts/model/estimator.py @@ -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