Files
aigc-apps-EasyAnimate/scripts/train_vae.sh
T
Bubbliiiingandzouxinyi0625 fb0916f4da Update EasyAnimateV2 (#5)
* update EasyAnimateV2

* update datasets loader

* update datasets loader

* update fast api

* complete data preprocess pipeline.

* update lots of readme

* update readme bans

* fix bug in validation while training

* provide example for video cut

* update text box

* add arxiv

* delete IDDPM

* update gallery

* update arxiv

* update readme

* update readme

* link fix

* update vae readme

---------

Co-authored-by: zouxinyi0625 <zouxinyi.zxy@alibaba-inc.com>
2024-05-31 10:33:14 +08:00

5 lines
249 B
Bash

torchrun --master_addr=127.0.0.1 --master_port=6006 --nnodes=1 --node_rank=0 --nproc_per_node=1 scripts/train_vae.py \
-t \
-b easyanimate/vae/configs/autoencoder/autoencoder_kl_32x32x4_slice.yaml \
--scale_lr=False \
--resume="True"