diff --git a/scripts/wan2.1_one_forcing/README_ONE_FORCING.md b/scripts/wan2.1_one_forcing/README_ONE_FORCING.md index d919f85..f10ffa2 100644 --- a/scripts/wan2.1_one_forcing/README_ONE_FORCING.md +++ b/scripts/wan2.1_one_forcing/README_ONE_FORCING.md @@ -188,8 +188,8 @@ accelerate launch --mixed_precision="bf16" --use_fsdp \ --num_train_epochs=100 \ --max_train_steps=200 \ --checkpointing_steps=100 \ - --learning_rate=1.0e-05 \ - --learning_rate_critic=1.0e-05 \ + --learning_rate=2.0e-06 \ + --learning_rate_critic=4.0e-07 \ --lr_scheduler="constant" \ --seed=42 \ --output_dir="output_dir_wan2.1_one_forcing_framewise_1step" \ @@ -239,7 +239,7 @@ The One-Forcing recipe sets these launcher flags: | `num_frame_per_block: 1` | `--num_frame_per_block` | `1` (frame-wise) | | `num_training_frames: 21` / `score_num_frames: 21` | `--video_sample_n_frames=81` @ 480×832 | 21 latent frames | | `guidance_scale: 5.0` | `--real_guidance_scale` | `6.0` (see CFG note below) | -| `lr: 1e-5` / `lr_critic: 1e-5` | `--learning_rate` / `--learning_rate_critic` | `1.0e-05` | +| `lr: 2e-6` / `lr_critic: 4e-7` (gen:critic=5:1) | `--learning_rate` / `--learning_rate_critic` | `2.0e-06` / `4.0e-07` | | `beta1: 0.0`, `beta2: 0.999`, `weight_decay: 0.01` | `--adam_beta1`, `--adam_beta2`, `--adam_weight_decay` | `0.0`, `0.999`, `0.01` | | `ema_weight: 0.99` | `--use_ema`, `--ema_decay` | on, `0.99` | | `max_steps: 200` | `--max_train_steps` | `200` | diff --git a/scripts/wan2.1_one_forcing/README_ONE_FORCING_zh-CN.md b/scripts/wan2.1_one_forcing/README_ONE_FORCING_zh-CN.md index 3688357..1554ce9 100644 --- a/scripts/wan2.1_one_forcing/README_ONE_FORCING_zh-CN.md +++ b/scripts/wan2.1_one_forcing/README_ONE_FORCING_zh-CN.md @@ -187,8 +187,8 @@ accelerate launch --mixed_precision="bf16" --use_fsdp \ --num_train_epochs=100 \ --max_train_steps=200 \ --checkpointing_steps=100 \ - --learning_rate=1.0e-05 \ - --learning_rate_critic=1.0e-05 \ + --learning_rate=2.0e-06 \ + --learning_rate_critic=4.0e-07 \ --lr_scheduler="constant" \ --seed=42 \ --output_dir="output_dir_wan2.1_one_forcing_framewise_1step" \ @@ -238,7 +238,7 @@ One-Forcing 配方设置以下 launcher 参数: | `num_frame_per_block: 1` | `--num_frame_per_block` | `1`(逐帧) | | `num_training_frames: 21` / `score_num_frames: 21` | `--video_sample_n_frames=81` @ 480×832 | 21 个 latent 帧 | | `guidance_scale: 5.0` | `--real_guidance_scale` | `6.0`(见下方 CFG 说明) | -| `lr: 1e-5` / `lr_critic: 1e-5` | `--learning_rate` / `--learning_rate_critic` | `1.0e-05` | +| `lr: 2e-6` / `lr_critic: 4e-7`(gen:critic=5:1) | `--learning_rate` / `--learning_rate_critic` | `2.0e-06` / `4.0e-07` | | `beta1: 0.0`、`beta2: 0.999`、`weight_decay: 0.01` | `--adam_beta1`、`--adam_beta2`、`--adam_weight_decay` | `0.0`、`0.999`、`0.01` | | `ema_weight: 0.99` | `--use_ema`、`--ema_decay` | 开启、`0.99` | | `max_steps: 200` | `--max_train_steps` | `200` | diff --git a/scripts/wan2.1_one_forcing/train_one_forcing.sh b/scripts/wan2.1_one_forcing/train_one_forcing.sh index 5509513..07a27a5 100644 --- a/scripts/wan2.1_one_forcing/train_one_forcing.sh +++ b/scripts/wan2.1_one_forcing/train_one_forcing.sh @@ -50,8 +50,8 @@ accelerate launch --mixed_precision="bf16" --use_fsdp \ --num_train_epochs=100 \ --max_train_steps=200 \ --checkpointing_steps=100 \ - --learning_rate=1.0e-05 \ - --learning_rate_critic=1.0e-05 \ + --learning_rate=2.0e-06 \ + --learning_rate_critic=4.0e-07 \ --lr_scheduler="constant" \ --seed=42 \ --output_dir="output_dir_wan2.1_one_forcing_framewise_1step" \