fix missing tb logs

ghstack-source-id: ac3501485faa093c8b9daacca9917805e2a987b7 Pull Request resolved: #389
pytorch · Jun 10, 2024 · 79b3d4c · 79b3d4c
1 parent 6de5d31
commit 79b3d4c
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):
     """