From 1fb23c6b3027759a95147142afc17e04fe6a04f5 Mon Sep 17 00:00:00 2001 From: "Dr. Kashif Rasul" Date: Sun, 8 Dec 2019 20:53:52 +0100 Subject: [PATCH] fix imports --- pts/model/deepar/deepar_estimator.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/pts/model/deepar/deepar_estimator.py b/pts/model/deepar/deepar_estimator.py index 176c085..e387b3e 100644 --- a/pts/model/deepar/deepar_estimator.py +++ b/pts/model/deepar/deepar_estimator.py @@ -3,6 +3,7 @@ from typing import List, Optional import numpy as np import torch +import torch.nn as nn from pts import Trainer from pts.feature import ( @@ -21,7 +22,7 @@ from pts.feature import ( InstanceSplitter, ) from pts.dataset import FieldName, ExpectedNumInstanceSampler -from pts.model import PTSEstimator +from pts.model import PTSEstimator, Predictor from pts.modules import DistributionOutput, StudentTOutput from .deepar_network import DeepARTrainingNetwork, DeepARPredictionNetwork @@ -186,7 +187,7 @@ class DeepAREstimator(PTSEstimator): embedding_dimension=self.embedding_dimension, lags_seq=self.lags_seq, scaling=self.scaling, - dtype=self.dtype).to(trained_network.device) + dtype=self.dtype) copy_parameters(trained_network, prediction_network)