Skip to content

Commit de868f9

Browse files
authored
add global seed during launch (#89)
1 parent c15d82b commit de868f9

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

train.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,7 @@ def main(args):
4949
# ==============================
5050
# Initialize Distributed Training
5151
# ==============================
52-
colossalai.launch_from_torch({})
52+
colossalai.launch_from_torch({}, seed=args.global_seed)
5353
coordinator = DistCoordinator()
5454
device = get_current_device()
5555

0 commit comments

Comments
 (0)