Skip to content

Commit 8646e14

Browse files
committed
update 1b confoigs
1 parent e996e85 commit 8646e14

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

configs/1B/H100.toml

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -3,13 +3,13 @@ project = "debug_1B_zero_band"
33
type_model = "llama2"
44

55
[train]
6-
micro_bs = 16
6+
micro_bs = 32
77
reshard_after_forward = true
88

99
[optim]
10-
batch_size = 2048
11-
warmup_steps = 500
10+
batch_size = 1024
11+
warmup_steps = 1000
1212
total_steps = 8192
1313

1414
[optim.optim]
15-
lr = 4e-4
15+
lr = 7e-4

0 commit comments

Comments
 (0)