fix missing tb logs

ghstack-source-id: ac3501485faa093c8b9daacca9917805e2a987b7 Pull Request resolved: pytorch#389
tianyu-l · Jun 10, 2024 · 32a9cef · 32a9cef
1 parent 1ff2a8a
commit 32a9cef
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/torchtitan/utils.py b/torchtitan/utils.py
@@ -50,6 +50,8 @@ def get_metrics_rank(world_mesh: DeviceMesh, parallel_dims: ParallelDims) -> int
     else:
         metrics_log_rank = 0
 
+    return metrics_log_rank
+
 
 def set_pg_timeouts(timeout, world_mesh):
     """