#!/bin/bash

#if using XLA
export XRT_WORKERS="localservice:0;grpc://localhost:40934"
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"

py=python3

#exp=transglower_residual_aistpp_expmap
#exp=moglow_aistpp_expmap
exp=$1
#seq_id=gKR_sFM_cAll_d28_mKR5_ch06
#seq_id=gLH_sFM_cAll_d16_mLH3_ch04
#seq_id=aistpp_gPO_sFM_cAll_d12_mPO4_ch19
#seq_id=aistpp_gMH_sFM_cAll_d22_mMH3_ch04
seq_id=$2
addr=$3
#seq_id=Streetdance_001
#seq_id=shadermotion_7
echo $exp $seq_id

# mkdir $addr/inference/generated/
# mkdir $addr/inference/generated/${exp}
# mkdir $addr/inference/generated/${exp}/predicted_mods
# mkdir $addr/inference/generated/${exp}/videos
mkdir /tmp/$addr/
mkdir /tmp/$addr/${exp}
mkdir /tmp/$addr/${exp}/predicted_mods
mkdir /tmp/$addr/${exp}/videos
#data_dir=data/aistpp_20hz
#data_dir=$SCRATCH/data/aistpp_20hz
#data_dir=$SCRATCH/data/dance_combined
data_dir=$SCRATCH/data/dance_combined_test
#data_dir=$SCRATCH/data/dance_combined2_test
#data_dir=$SCRATCH/data/dance_combined2
#data_dir=test_data
#data_dir=$SCRATCH/data/moglow_pos

# if we don't pass seq_id it will choose a random one from the test set
$py inference/generate.py --data_dir=$data_dir --output_folder=/tmp/$addr --experiment_name=$exp \
    --seq_id $seq_id \
    ${@:4}
    #--generate_video \