CRSArena / script /opendialkg /ask_unicrs.sh
Nolwenn
Initial commit
b599481
raw
history blame
431 Bytes
CUDA_VISIBLE_DEVICES=0 python ask.py --api_key your_api_key --crs_model unicrs --dataset opendialkg_eval --kg_dataset opendialkg --model microsoft/DialoGPT-small --rec_model utils/model/unicrs_rec_opendialkg/ --conv_model utils/model/unicrs_conv_opendialkg/ --context_max_length 128 --entity_max_length 20 --tokenizer_path microsoft/DialoGPT-small --text_tokenizer_path roberta-base --resp_max_length 64 --text_encoder roberta-base