meng2003/music2dance

1#!/bin/bash2 3#export TPU_IP_ADDRESS=10.8.195.90;4#export XRT_TPU_CONFIG="tpu_worker;0;$TPU_IP_ADDRESS:8470"5#export TPU_NAME="grpc://$TPU_IP_ADDRESS:8470"6export XRT_WORKERS="localservice:0;grpc://localhost:40934"7export XRT_DEVICE_MAP="CPU:0;/job:localservice/replica:0/task:0/device:XLA_CPU:0|GPU:0;/job:localservice/replica:0/task:0/device:XLA_GPU:0"8#export PYTHONPATH=$SCRATCH/:${PYTHONPATH}9#export PYTHONPATH=/gpfsscratch/rech/imi/usc19dv/lib/python3.7/site-packages:${PYTHONPATH}10module load pytorch-gpu/py3/1.8.011 12py=python313 14#root_dir=$SCRATCH/data15root_dir=data16 17####aistpp_60hz18#data_dir=${root_dir}/scaled_features19#hparams_file=aistpp_60hz/transflower_aistpp_expmap20#hparams_file=aistpp_60hz/transglower_aistpp_expmap21 22####aistpp_20hz23#data_dir=${root_dir}/aistpp_20hz24#exp=$125#exp=transglower_aistpp_expmap26#exp=transglower_residual_aistpp_expmap27#exp=transflower_residual_aistpp_expmap28#exp=mowgli_aistpp_expmap29#exp=transflower_aistpp_expmap30#exp=transflower_residual_aistpp_expmap31#exp=residualflower2_transflower_aistpp_expmap32#exp=moglow_aistpp_expmap33#hparams_file=aistpp_20hz/${exp}34#exp=mowgli_aistpp_expmap_future335 36## Fix: needs vmapped version of transformer:37#hparams_file=aistpp_20hz/residualflower2_moglow_aistpp_expmap38 39####moglow_pos40#data_dir=${root_dir}/moglow_pos41#exp=$142#exp=transglower_moglow_pos43#exp=transglower_residual_moglow_pos44#exp=transflower_residual_moglow_pos45#exp=transflower_moglow_pos46#exp=residualflower2_transflower_moglow_pos47#exp=moglow_moglow_pos48#exp=moglow_trans_moglow_pos49#hparams_file=moglow_pos/${exp}50#exp=testing51#exp=${exp}_pos_emb52 53####dance_combined54#data_dir=${root_dir}/dance_combined55data_dir=${root_dir}/dance_combined256#data_dir=${root_dir}/dance_combined357exp=$158#exp=transformer_expmap59#exp=mowgli_expmap_stage260#exp=mowgli_expmap61#exp=transflower_expmap_cr462#exp=transglower_aistpp_expmap63#exp=transglower_residual_aistpp_expmap64#exp=transflower_residual_aistpp_expmap65#exp=transflower_aistpp_expmap66#exp=residualflower2_transflower_aistpp_expmap67#exp=moglow_aistpp_expmap68hparams_file=dance_combined/${exp}69 70#exp=${exp}_future3_actnorm71#exp=${exp}_future372exp=testing73#exp=mowgli_expmap_stage2_newdata274#exp=testing275 76echo $exp77 78$py training/train.py --data_dir=${data_dir} --max_epochs=2000\79    --hparams_file=training/hparams/${hparams_file}.yaml \80    --experiment_name=$exp\81    --accelerator=ddp \82    --workers=0 \83    --gpus=1 \84    ${@:2} #NOTE: can override experiment_name, and any of the options above85    #--batch_size=8 \86    #--continue_train \87    #--no_load_hparams \88    #--load_weights_only \89    #--only_load_in_state_dict=vae \90    #--fix_lengths \91    #--continue_train \92    #--workers=$(nproc) \93    #--gpus=1 \94    #--scales="[[16,0]]" \95    #--flow_dist=studentT \96    #--output_lengths="3" \97#    --stage2 \98#    --continue_train \99#    --load_weights_only \100#    --prior_use_x_transformers \101#    --use_x_transformers \102#    --use_rotary_pos_emb \103#    --learning_rate=1e-5 \104    #--residual_scales="[[16,0]]"105#    --glow_norm_layer="actnorm" \106    #--use_pos_emb_output \107#    --tpu_cores=8 \108