small update for custom models with dropout

OMalenfantThuot · Dec 14, 2020 · 398e8d4 · 398e8d4
1 parent 405ecb6
commit 398e8d4
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/mlcalcdriver/calculators/schnetpack.py b/mlcalcdriver/calculators/schnetpack.py
@@ -54,6 +54,7 @@ def __init__(
                 os.environ["MODELDIR"] + model_dir, map_location=self.device
             )
         super(SchnetPackCalculator, self).__init__(units=units)
+        self.model.eval()
         self._get_representation_type()
 
     @property