aistpp_pytorch

Learn to Dance with AIST++: Music Conditioned 3D Dance Generation.

Installation

pip install -r requirements.txt

Prepare Dataset

bash data/scripts/prepare_dataset.sh

Train

python train.py --gpus 4 \
                --max_epochs 60000 \
                --lr 3e-5 \
                --lr_decay_milestones 36000 48000 \
                --lr_warmup

Inference

python inference.py --ckpt path/to/your/checkpoint

GitHub

https://github.com/serend1p1ty/aistpp_pytorch