From 5c27d32776351526e94d23c55200e938aff5aeb8 Mon Sep 17 00:00:00 2001 From: Pedro Cuenca Date: Fri, 24 Feb 2023 13:25:33 +0000 Subject: [PATCH] Attempt to unwrap tracker. --- src/accelerate/accelerator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/accelerate/accelerator.py b/src/accelerate/accelerator.py index 0177341e906..152efd6ccd0 100644 --- a/src/accelerate/accelerator.py +++ b/src/accelerate/accelerator.py @@ -1936,7 +1936,7 @@ def get_tracker(self, name: str, unwrap: bool = False): if len(getattr(self, "trackers", [])) > 0: for tracker in self.trackers: if tracker.name == name: - return tracker + return tracker.tracker if unwrap else tracker raise ValueError(f"{name} is not an available tracker stored inside the `Accelerator`.") # Handle tracker only made on main process return GeneralTracker(_blank=True)