diff --git a/README.md b/README.md index 2a24d90..88994c4 100644 --- a/README.md +++ b/README.md @@ -78,7 +78,7 @@ Before starting the training, please check whether the dataset has been prepared ```bash LEARNING_RATES=("1e-4" "1e-3") LR_SCHEDULES=("cosine_with_restarts") - OPTIMIZERS=("adamw", "adam") + OPTIMIZERS=("adamw" "adam") MAX_TRAIN_STEPS=("3000") ``` diff --git a/README_zh.md b/README_zh.md index 689e6bd..4bc11b2 100644 --- a/README_zh.md +++ b/README_zh.md @@ -77,7 +77,7 @@ diffusers(该分支为 CogVideoX 的图像到视频添加了 LoRA 加载支持 ``` LEARNING_RATES=("1e-4" "1e-3") LR_SCHEDULES=("cosine_with_restarts") - OPTIMIZERS=("adamw", "adam") + OPTIMIZERS=("adamw" "adam") MAX_TRAIN_STEPS=("3000") ``` diff --git a/train_image_to_video_lora.sh b/train_image_to_video_lora.sh index 565f31a..c219e8c 100755 --- a/train_image_to_video_lora.sh +++ b/train_image_to_video_lora.sh @@ -10,7 +10,7 @@ GPU_IDS="0" # Experiment with as many hyperparameters as you want! LEARNING_RATES=("1e-4" "1e-3") LR_SCHEDULES=("cosine_with_restarts") -OPTIMIZERS=("adamw", "adam") +OPTIMIZERS=("adamw" "adam") MAX_TRAIN_STEPS=("3000") # Single GPU uncompiled training diff --git a/train_text_to_video_lora.sh b/train_text_to_video_lora.sh index 2892c75..4660535 100755 --- a/train_text_to_video_lora.sh +++ b/train_text_to_video_lora.sh @@ -10,7 +10,7 @@ GPU_IDS="0" # Experiment with as many hyperparameters as you want! LEARNING_RATES=("1e-4" "1e-3") LR_SCHEDULES=("cosine_with_restarts") -OPTIMIZERS=("adamw", "adam") +OPTIMIZERS=("adamw" "adam") MAX_TRAIN_STEPS=("3000") # Single GPU uncompiled training