| CUDA_VISIBLE_DEVICES=2 python main_nba_mid_graphv6_v3.py \ | |
| --data_dir /mnt/jaewoo4tb/srtp/MoFlow/data/nba/original \ | |
| --exp_name mid_nba_graphv6v3_fixed \ | |
| --epochs 100 \ | |
| --batch_size 32 \ | |
| --lr 1e-3 \ | |
| --eval_every 5 \ | |
| --sampling ddim \ | |
| --sampling_step 10 |