From 0a8e9fdec8549a97a501e0a21b820925498e8ff4 Mon Sep 17 00:00:00 2001 From: Kashif Rasul Date: Mon, 13 Jan 2020 20:01:51 +0100 Subject: [PATCH] set prediction network to eval mode --- pts/model/predictor.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/pts/model/predictor.py b/pts/model/predictor.py index 0874f84..39a6e77 100644 --- a/pts/model/predictor.py +++ b/pts/model/predictor.py @@ -59,6 +59,9 @@ class PTSPredictor(Predictor): device=self.device, dtype=self.dtype, ) + + self.prediction_net.eval() + with torch.no_grad(): yield from self.forecast_generator( inference_data_loader=inference_data_loader,