From 7419829ebc00bc16b7b4526ed04b0e69756b59e6 Mon Sep 17 00:00:00 2001 From: SolitaryThinker Date: Sun, 22 Jun 2025 04:19:10 -0700 Subject: [PATCH] update --- .../finetune/wan_i2v_14b_480p/crush_smol/finetune.slurm | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/training/finetune/wan_i2v_14b_480p/crush_smol/finetune.slurm b/examples/training/finetune/wan_i2v_14b_480p/crush_smol/finetune.slurm index 929155239..c4a4dd622 100644 --- a/examples/training/finetune/wan_i2v_14b_480p/crush_smol/finetune.slurm +++ b/examples/training/finetune/wan_i2v_14b_480p/crush_smol/finetune.slurm @@ -72,10 +72,10 @@ srun torchrun \ --train_sp_batch_size 1\ --dataloader_num_workers 10\ --gradient_accumulation_steps=2\ - --max_train_steps=5000 \ + --max_train_steps=10000 \ --learning_rate=5e-5\ --mixed_precision="bf16"\ - --checkpointing_steps=6000 \ + --checkpointing_steps=11000 \ --validation_steps 100\ --validation_sampling_steps "40" \ --log_validation \