diff --git a/scripts/experiment_11.sh b/scripts/experiment_11.sh index 151f43aef..375cf1cfe 100644 --- a/scripts/experiment_11.sh +++ b/scripts/experiment_11.sh @@ -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\ diff --git a/scripts/experiment_12.sh b/scripts/experiment_12.sh index 4cec6dcc0..2f17d8ad4 100644 --- a/scripts/experiment_12.sh +++ b/scripts/experiment_12.sh @@ -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\