Multiple improvements

MGAMZ · Jan 17, 2025 · 8f37dd2 · 8f37dd2
1 parent 861fc1b
commit 8f37dd2
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 4 deletions.
diff --git a/mmengine/model/averaged_model.py b/mmengine/model/averaged_model.py
@@ -103,7 +103,6 @@ def update_parameters(self, model: nn.Module) -> None:
             for k, p_avg in self.avg_parameters.items():
                 p_avg.data.copy_(src_parameters[k].data)
         elif self.steps % self.interval == 0:
-            print(self.avg_parameters)
             for k, p_avg in self.avg_parameters.items():
                 if p_avg.dtype.is_floating_point:
                     device = p_avg.device

diff --git a/mmengine/runner/loops.py b/mmengine/runner/loops.py
@@ -418,9 +418,9 @@ def run_iter(self, idx, data_batch: Sequence[dict]):
         # outputs should be sequence of BaseDataElement
         with autocast(enabled=self.fp16):
             outputs = self.runner.model.val_step(data_batch)
-
+        
         outputs, self.val_loss = _update_losses(outputs, self.val_loss)
-
+        
         self.evaluator.process(data_samples=outputs, data_batch=data_batch)
         self.runner.call_hook(
             'after_val_iter',

diff --git a/mmengine/visualization/vis_backend.py b/mmengine/visualization/vis_backend.py
@@ -578,7 +578,7 @@ def add_image(self,
                   step: int = 0,
                   **kwargs) -> None:
         """Record the image to tensorboard.
-
+        
         Args:
             name (str): The image identifier.
             image (np.ndarray): The image to be saved. The format