#!/bin/bash

set -x
ulimit -c 0

set -x
ulimit -c 0
export WANDB_INIT_TIMEOUT=200

script_name=${1}
dataset=${2}

echo "--------------------------------------------------" >> ~/history.txt
echo "Slurm job id | job id | command | model | dataset" >> ~/history.txt
echo "$SLURM_JOB_ID | $JOB_ID | evaluation | $script_name | $dataset" >> ~/history.txt
WANDB_KEY=4c1540ebf8cb9964703ac212a937c00848a79b67
wandb login ${WANDB_KEY}
export WANDB__SERVICE_WAIT=300

# evaluate
CUDA_VISIBLE_DEVICES=0 python genie/evaluate.py --checkpoint_dir data/${script_name}  \
    --val_data_dir data/${dataset}_magvit_traj1000000_val   --save_outputs_dir data/${script_name}_${dataset} &

# CUDA_VISIBLE_DEVICES=1 python genie/evaluate.py --checkpoint_dir data/${script_name}  \
#     --val_data_dir data/${dataset}_magvit_traj1000000_val --autoregressive_time  --add_action_input  --save_outputs_dir data/${script_name}_${dataset}

mkdir -p data/$script_name_${dataset}/output


python genie/generate.py   --checkpoint_dir data/${script_name}  \
--val_data_dir data/${dataset}_magvit_traj1000000_val  --teacher_force_time \
--output_dir data/${script_name}_${dataset}/output

# visualize
python visualize.py --token_dir data/${script_name}_${dataset}/output