diff --git a/src/modalities/evaluator.py b/src/modalities/evaluator.py
index 0c6b31ed..a5ed5668 100644
--- a/src/modalities/evaluator.py
+++ b/src/modalities/evaluator.py
@@ -104,6 +104,9 @@ def evaluate(
                 evaluation_result=evaluation_result,
             )
             result_dict[data_loader.dataloader_tag] = evaluation_result
+
+        model.train()
+
         return result_dict
 
     @staticmethod
diff --git a/src/modalities/trainer.py b/src/modalities/trainer.py
index c77f45ba..ee609a7d 100644
--- a/src/modalities/trainer.py
+++ b/src/modalities/trainer.py
@@ -181,7 +181,6 @@ def train(
                 )
                 thoughput_aggregator.remove_keys()
 
-                model.train()
                 cumulated_losses = self._reset_tracked_losses()
 
             evaluation_callback(train_step_id=train_step_id)