Skip to content

Commit

Permalink
add simple 10b configs
Browse files Browse the repository at this point in the history
  • Loading branch information
samsja committed Dec 11, 2024
1 parent a974cf5 commit 337f731
Showing 1 changed file with 23 additions and 0 deletions.
23 changes: 23 additions & 0 deletions configs/10B/H100_simple.toml
Original file line number Diff line number Diff line change
@@ -0,0 +1,23 @@
name_model = "10B"
project = "debug_10B_zero_band"

[train]
micro_bs = 1
ac_ckpt = true

[optim]
sched_type = "wsd-sqrt"
batch_size = 128 #1M tokens bs
warmup_steps = 1000
total_steps = 1_000_000_000_000
lr = 7.5e-5

adam_betas1 = 0.9
adam_betas2 = 0.95
weight_decay = 0.1

z_loss = true

[data]
seq_length = 8192
num_workers = 4

0 comments on commit 337f731

Please sign in to comment.