fix another bug in eval method

mu_map/training/lib.py

@@ -112,8 +112,8 @@ class AbstractTraining:
 
     def _eval_epoch(self):
         torch.set_grad_enabled(False)
-        for model in self.models:
-            model.eval()
+        for param in self.training_params:
+            param.model.eval()
 
         loss = 0.0
         data_loader = self.data_loaders["validation"]