Spaces:
Runtime error
Runtime error
experiment=$1 | |
checkpoint=$2 | |
if [[ "$experiment" == "SGD" ]]; then | |
echo "Conduct experiment with SGD dataset" | |
job_name='SGD-full' | |
data_list="sgd" # 165k training examples | |
eval_interval=50000 # evaluation interval | |
elif [[ "$experiment" == "MultiWOZ" ]]; then | |
echo "Conduct experiment with MulwiWOZ dataset" | |
job_name='MultiWOZ-full' | |
data_list="multiwoz" # 56k training examples | |
eval_interval=20000 | |
elif [[ "$experiment" == "Joint" ]]; then | |
echo "Conduct experiment with SGD + MulwiWOZ dataset" | |
job_name='Joint-full' | |
data_list="sgd multiwoz" # 221k training examples | |
eval_interval=70000 | |
else | |
echo "Unrecognised argument" | |
exit | |
fi | |
mkdir -p log decode | |
decode_file='decode/'$job_name'.json' | |
eye_browse_output=true # set to false for storing generation results in file | |
python main.py --mode='testing' \ | |
--model_name=$job_name \ | |
--checkpoint=$checkpoint \ | |
--decode_file=$decode_file \ | |
--data_dir="processed_data" \ | |
--data_list=$data_list \ | |
--eye_browse_output=$eye_browse_output | |