meng2003/music2dance

1#!/bin/bash2 3export TPU_IP_ADDRESS=10.104.22.146;4#export TPU_IP_ADDRESS=10.95.66.34;5export XRT_TPU_CONFIG="tpu_worker;0;$TPU_IP_ADDRESS:8470"6export TPU_NAME="grpc://$TPU_IP_ADDRESS:8470"7#export XRT_WORKERS="localservice:0;grpc://localhost:40934"8#export 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"9#export PYTHONPATH=$SCRATCH/:${PYTHONPATH}10#export PYTHONPATH=/gpfsscratch/rech/imi/usc19dv/lib/python3.7/site-packages:${PYTHONPATH}11 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_20hz23data_dir=${root_dir}/aistpp_20hz24#exp=$125exp=testing26#exp=transglower_aistpp_expmap27#exp=transglower_residual_aistpp_expmap28#exp=transflower_residual_aistpp_expmap29#exp=transflower_aistpp_expmap30#exp=residualflower2_transflower_aistpp_expmap31#exp=moglow_aistpp_expmap32#hparams_file=aistpp_20hz/${exp}33hparams_file=aistpp_20hz/mowgli_aistpp_expmap_testing34 35## Fix: needs vmapped version of transformer:36#hparams_file=aistpp_20hz/residualflower2_moglow_aistpp_expmap37 38####dance_combined39#data_dir=${root_dir}/dance_combined40#exp=$141#exp=transflower_expmap42#exp=moglow_expmap43#hparams_file=dance_combined/${exp}44 45#exp=${exp}_future3_actnorm46#exp=${exp}_future347#exp=${exp}_future348 49echo $exp50 51$py training/train.py --data_dir=${data_dir} --max_epochs=1000\52    --model=mowgli2 \53    --do_validation \54    --val_batch_size=32 \55    --batch_size=32 \56    --experiment_name=$exp\57    --workers=$(nproc) \58    --tpu_cores=8 \59    --hparams_file=training/hparams/${hparams_file}.yaml \60    #--continue_train \61    #--load_weights_only \62    #--stage2 \63    #--prior_use_x_transformers \64    #--output_lengths="3" \65    #--max_prior_loss_weight=0.01 \66    #--accelerator=ddp \67    #--scales="[[16,0]]" \68#    --use_rotary_pos_emb \69    #--residual_scales="[[16,0]]"70#    --glow_norm_layer="actnorm" \71    #--use_pos_emb_output \72    #--gpus=2 \73