wassname/pytorch-ts
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-ts.git synced 2026-08-13 12:30:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix device

Browse Source
This commit is contained in:
Kashif Rasul
2019-11-18 23:31:05 +01:00
parent db783e2005
commit 563ef86436
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pts/model/estimator.py
+1 -1
View File
@@ -108,7 +108,7 @@ class PTSEstimator(Estimator):
)
# ensure that the training network is created on the same device
trained_net = self.create_training_network(self.device)
trained_net = self.create_training_network(self.trainer.device)
self.trainer(
net=trained_net,