Skip to content

Commit

Permalink
fix config file
Browse files Browse the repository at this point in the history
  • Loading branch information
ahmeda14960 committed Oct 17, 2024
1 parent 667a5a3 commit 37e77fb
Showing 1 changed file with 0 additions and 6 deletions.
6 changes: 0 additions & 6 deletions config/llama_7b_with_olmo_config.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -15,10 +15,6 @@ trainer:
project: "marin"
tags: ["dolma", "olmo", "llama"]

checkpointer:
keep:
- every: 250

mp: p=f32,c=bfloat16
train_batch_size: 2048
num_train_steps: 750000 # 3,000,000,000,000 / 4,000,000 = 750,000
Expand All @@ -31,5 +27,3 @@ optimizer:
weight_decay: 0.1
min_lr_ratio: 0.1
warmup: 0.01

data_shuffle: true

0 comments on commit 37e77fb

Please sign in to comment.