AuRoRA / run_correction.sh
Anni123's picture
Upload folder using huggingface_hub
b6a1d8d
raw
history blame
414 Bytes
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