copy trainer network to device

2026-08-11 05:51:21 +08:00 · 2020-01-28 12:12:32 +01:00
parent 23b0d39412
commit 438785cde1
1 changed files with 1 additions and 1 deletions
@@ -215,7 +215,7 @@ class TransformerEstimator(PTSEstimator):
            lags_seq=self.lags_seq,
            scaling=self.scaling,
            num_parallel_samples=self.num_parallel_samples,
-        )
+        ).to(device)

        copy_parameters(trained_network, prediction_network)