deepgeek/MuseTalk
2
0
Fork 0
mirror of https://github.com/TMElyralab/MuseTalk.git synced 2026-02-05 09:59:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

fixed mltiple video data preperation

Browse Source
This commit is contained in:
Shounak Banerjee
2024-06-17 18:39:15 +00:00
parent af82f3b00f
commit b968548131
3 changed files with 15 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
train_codes/train.sh
View File

@@ -7,13 +7,12 @@ accelerate launch train.py \
--unet_config_file=$UNET_CONFIG \
--pretrained_model_name_or_path=$VAE_MODEL \
--data_root=$DATASET \
--train_batch_size=8 \
--gradient_accumulation_steps=4 \
--train_batch_size=256 \
--gradient_accumulation_steps=16 \
--gradient_checkpointing \
--max_train_steps=100000 \
--learning_rate=5e-05 \
--max_grad_norm=1 \
--lr_scheduler="cosine" \
--lr_warmup_steps=0 \
--output_dir="output" \
--val_out_dir='val' \
@@ -25,5 +24,6 @@ accelerate launch train.py \
--use_audio_length_left=2 \
--use_audio_length_right=2 \
--whisper_model_type="tiny" \
--train_json="/root/MuseTalk/train.json" \
--val_json="/root/MuseTalk/val.json" \
--train_json="../train.json" \
--val_json="../val.json" \
--lr_scheduler="cosine" \