wassname/pytorch-ts
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-ts.git synced 2026-06-27 19:32:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

move predictor network to device

Browse Source
This commit is contained in:
Dr. Kashif Rasul
2020-02-12 17:18:02 +01:00
parent 1f3c2ef943
commit 923ba8978a
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pts/model/simple_feedforward/simple_feedforward_estimator.py
+1 -1
View File
@@ -158,7 +158,7 @@ class SimpleFeedForwardEstimator(PTSEstimator):
batch_normalization=self.batch_normalization,
mean_scaling=self.mean_scaling,
num_parallel_samples=self.num_parallel_samples,
)
).to(device)
copy_parameters(trained_network, prediction_network)