Skip to content

Commit 5951c9a

Browse files
authored
fix: Rank is set to zero by default (#594)
Signed-off-by: Padmanabha V Seshadri <[email protected]>
1 parent ef11bd1 commit 5951c9a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tuning/trainercontroller/controllermetrics/per_process_state.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -74,4 +74,4 @@ def compute(self, _: TrainerState = None, **kwargs) -> Any:
7474
"""
7575
if torch.distributed.is_available() and torch.distributed.is_initialized():
7676
return {"rank": torch.distributed.get_rank()}
77-
return {"rank": None}
77+
return {"rank": 0}

0 commit comments

Comments
 (0)