obey min num frames requirement in self forcing test

This commit is contained in:
RandNMR73
2025-10-02 09:35:31 +00:00
parent be01fbe536
commit a274a90c88
@@ -37,7 +37,7 @@ def run_worker():
"--data_path", "data/crush-smol_processed_t2v/combined_parquet_dataset",
"--validation_dataset_file", "examples/training/finetune/wan_t2v_1.3B/crush_smol/validation.json",
"--train_batch_size", "1",
"--num_latent_t", "6",
"--num_latent_t", "21",
"--num_gpus", "2",
"--sp_size", "1",
"--tp_size", "1",
@@ -62,7 +62,7 @@ def run_worker():
"--wandb_run_name", wandb_name,
"--num_height", "480",
"--num_width", "832",
"--num_frames", "6",
"--num_frames", "21",
"--flow_shift", "5",
"--validation_guidance_scale", "1.0",
"--weight_decay", "0.01",
@@ -83,7 +83,7 @@ def run_worker():
"--simulate_generator_forward",
"--num_frame_per_block", "3",
"--enable_gradient_masking",
"--gradient_mask_last_n_frames", "6",
"--gradient_mask_last_n_frames", "21",
"--independent_first_frame", "False",
"--same_step_across_blocks", "True",
"--last_step_only", "False",