From 7a6dff2d9fa08e989f086978ddf564f1e07e5fba Mon Sep 17 00:00:00 2001 From: Johan Mellin Date: Mon, 21 Oct 2024 00:30:51 +0200 Subject: [PATCH] Fixed optimizers parsing error in bash scripts (#61) * replaced lambda statements with local functions to work with pickle in dataset.py and moved collate function outside of main and added its own class in cogvideox_image_to_video_lora.py * fixed merge issue * fixed row deletion merge * fixed row deletion merge * fixed comma-separated param issue in bash --- README.md | 2 +- README_zh.md | 2 +- train_image_to_video_lora.sh | 2 +- train_text_to_video_lora.sh | 2 +- 4 files changed, 4 insertions(+), 4 deletions(-) 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