Spaces:
Running
on
Zero
Running
on
Zero
File size: 1,240 Bytes
246c106 |
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 |
#!/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
|