We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent b3638d9 commit 1d81b1fCopy full SHA for 1d81b1f
1 file changed
recipes/esm2_native_te_nvfsdp_thd/hydra_config/L1_3B_ddp.yaml
@@ -2,7 +2,7 @@ defaults:
2
- defaults
3
4
# Training config
5
-model_name: esm2_t33_650M_UR50D
+model_name: esm2_t36_3B_UR50D
6
micro_batch_size: 32
7
num_train_steps: 10_000
8
0 commit comments