diff --git a/composer/trainer/trainer.py b/composer/trainer/trainer.py index fe021df727..ef74c2e5ba 100644 --- a/composer/trainer/trainer.py +++ b/composer/trainer/trainer.py @@ -2246,14 +2246,12 @@ def _ensure_metrics_device_and_dtype( return metrics def _compute_and_log_metrics(self, dataloader_label: str, metrics: Dict[str, Metric]): - """Computes metrics, logs the results, and updates the state with the deep-copied metrics. + """Computes metrics, logs the results, and updates the state with the metrics. Args: dataloader_label (str): The dataloader label. metrics (Dict[str, Metric]): The metrics to compute. """ - metrics = deepcopy(metrics) - # log computed metrics computed_metrics = {} for metric_name, metric in metrics.items():