Spaces:
Sleeping
Sleeping
File size: 1,359 Bytes
bfbfae6 5bb52cc bfbfae6 5bb52cc bfbfae6 5bb52cc bfbfae6 5bb52cc bfbfae6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 |
#!/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 \
|