From b3d16e02845fcb45848815d261a09b04bd1ff19e Mon Sep 17 00:00:00 2001 From: rlsu9 Date: Sun, 1 Dec 2024 07:58:24 +0400 Subject: [PATCH] typo --- scripts/experiment_11.sh | 2 +- scripts/experiment_12.sh | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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\