Skip to content

Commit d1399cc

Browse files
pstjohnpstjohn
committed
fix 3b config
Signed-off-by: Peter St. John <pstjohn@nvidia.com>
1 parent be8da13 commit d1399cc

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

  • recipes/esm2_native_te_mfsdp/hydra_config

recipes/esm2_native_te_mfsdp/hydra_config/L1_3B_ddp.yaml renamed to recipes/esm2_native_te_mfsdp/hydra_config/L1_3B.yaml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,8 +2,8 @@ defaults:
22
- defaults
33

44
# Training config
5-
model_name: nvidia/esm2_t33_650M_UR50D
6-
micro_batch_size: 32
5+
model_name: nvidia/esm2_t36_3B_UR50D
6+
micro_batch_size: 16
77
num_train_steps: 10_000
88

99
# WandB config

0 commit comments

Comments
 (0)