File size: 300 Bytes
d4cbafd | 1 2 3 4 5 6 7 8 9 | 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 |