update

utils/model.py

@@ -40,6 +40,7 @@ class Model(torch.nn.Module):
 
         logging.info(f'Loaded model {self.name}')
 
+        self.model.eval()
         self.update()
 
     @classmethod