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 | |