This commit is contained in:
Peiyuan Zhang
2024-12-16 04:43:53 +00:00
parent 50b92fded2
commit ff5d9f5191
2 changed files with 2 additions and 2 deletions
+1 -1
View File
@@ -35,7 +35,7 @@ torchrun --nnodes 4 --nproc_per_node 8\
--mixed_precision="bf16"\
--checkpointing_steps=64\
--validation_steps 64\
--validation_sampling_steps "2,4,8" \
--validation_sampling_steps "4,6,8" \
--checkpoints_total_limit 3\
--allow_tf32\
--ema_start_step 0\
+1 -1
View File
@@ -35,7 +35,7 @@ torchrun --nnodes 4 --nproc_per_node 8\
--mixed_precision="bf16"\
--checkpointing_steps=64\
--validation_steps 64\
--validation_sampling_steps "2,4,8" \
--validation_sampling_steps "4,6,8" \
--checkpoints_total_limit 3\
--allow_tf32\
--ema_start_step 0\