export OPENAI_API_KEY="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" | |
DATASET="strategyqa" | |
LIMIT_DATASET_SIZE="1" | |
DEMO_PATH="./demos/${DATASET}" | |
OUTPUT_DIR="./experiment/${DATASET}" | |
RETR_METHOD="kb_only" | |
python self_correction.py \ | |
--dataset ${DATASET} \ | |
--limit_dataset_size ${LIMIT_DATASET_SIZE} \ | |
--demo_path ${DEMO_PATH} \ | |
--output_dir ${OUTPUT_DIR} \ | |
--retrieve_method ${RETR_METHOD} \ | |
--resume_id 45 |