Skip to content

Commit c4e6458

Browse files
author
Thomas Capelle
committed
rank naming
1 parent fe73982 commit c4e6458

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

train.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -227,6 +227,7 @@ def train_model(config: ModelConfig):
227227
# set the wandb project where this run will be logged
228228
project="pytorch-transformer-distributed",
229229
# allow resuming existing run with the same name (in case the rank 0 node crashed)
230+
name=f"global_rank_{config.global_rank}",
230231
id=wandb_run_id,
231232
resume="allow",
232233
group=config.wandb_group,

0 commit comments

Comments
 (0)