set prediction network to eval mode

2026-09-09 11:32:17 +08:00 · 2020-01-13 20:01:51 +01:00
parent 8cb33f20c6
commit 0a8e9fdec8
1 changed files with 3 additions and 0 deletions
@@ -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,