typo

2024-12-01 07:58:24 +04:00
parent c466fe9d73
commit b3d16e0284
2 changed files with 2 additions and 2 deletions
@@ -20,7 +20,7 @@ torchrun --nnodes 2 --nproc_per_node 8\
    --gradient_checkpointing\
    --train_batch_size=1\
    --num_latent_t 28\
-    --sp_size 2\
+    --sp_size 4\
    --train_sp_batch_size 2\
    --dataloader_num_workers 4\
    --gradient_accumulation_steps=1\
@@ -20,7 +20,7 @@ torchrun --nnodes 2 --nproc_per_node 8\
    --gradient_checkpointing\
    --train_batch_size=1\
    --num_latent_t 28\
-    --sp_size 2\
+    --sp_size 4\
    --train_sp_batch_size 2\
    --dataloader_num_workers 4\
    --gradient_accumulation_steps=1\