From 84cf3b67954b84a14e87823609c8d40ba779ca4e Mon Sep 17 00:00:00 2001 From: Peiyuan Zhang Date: Fri, 25 Oct 2024 09:58:41 +0000 Subject: [PATCH] Update t2v_debug_multi.sh with video_length_tolerance_range and dataloader_num_workers --- t2v_debug_multi.sh | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/t2v_debug_multi.sh b/t2v_debug_multi.sh index 5cf73a7f6..7d2e7abbb 100644 --- a/t2v_debug_multi.sh +++ b/t2v_debug_multi.sh @@ -14,6 +14,7 @@ accelerate launch \ --sample_rate 1 \ --num_frames 29 \ --train_fps 24 \ + --video_length_tolerance_range 200 \ --max_height 480 \ --max_width 640 \ --interpolation_scale_t 1.0 \ @@ -22,7 +23,7 @@ accelerate launch \ --attention_mode xformers \ --gradient_checkpointing \ --train_batch_size=1 \ - --dataloader_num_workers 10 \ + --dataloader_num_workers 1 \ --gradient_accumulation_steps=1 \ --max_train_steps=1000000 \ --learning_rate=1e-4 \