# Run code
CUDA_VISIBLE_DEVICES=GPU_NUMBER \
HYDRA_FULL_ERROR=1 \
python3 train.py \
name="Aist++_M2D_baseline" \
server_name=SERVER_NAME \
gpu_name=GPU_NAME \
experiment=AISTPP \
trainer.min_epochs=1 \
trainer.max_epochs=6000 \
datamodule.batch_size=16 \
callbacks.model_checkpoint.every_n_train_steps=100000 \
model.gen_params.reconstruction=True \
model.loss_params.alphas=[0.636,2.964]
-
Notifications
You must be signed in to change notification settings - Fork 0
JiHyuk-Byun/Music2Dance_Transformer
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
About
No description, website, or topics provided.
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published