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