forked from ZonglinY/MOOSE
-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.sh
18 lines (16 loc) · 930 Bytes
/
main.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
#!/bin/bash
#SBATCH -J baseline
#SBATCH --partition=DGXq
#SBATCH -w node18
#SBATCH --gres=gpu:1
#SBATCH --output /export/home/zonglin001/Outs/Tomato/claude_50bkg_0itr_bkgnoter0_indirect0_onlyindirect0_close0_ban1_baseline2_survey1_bkgInspPasgSwap0_hypSuggestor0.out
# vicuna / gpt2 / chatgpt / vicuna13 / falcon / claude
python -u main.py --model_name claude \
--output_dir ~/Checkpoints/Tomato/claude_50bkg_0itr_bkgnoter0_indirect0_onlyindirect0_close0_ban1_baseline2_survey1_bkgInspPasgSwap0_hypSuggestor0 \
--num_background_for_hypotheses 50 --num_CoLM_feedback_times 0 --bkg_corpus_chunk_noter 0 \
--if_indirect_feedback 0 --if_only_indirect_feedback 0 \
--if_close_domain 0 --if_ban_selfeval 1 \
--if_baseline 2 \
--if_novelty_module_have_access_to_surveys 1 --if_insp_pasg_for_bkg_and_bkg_pasg_included_in_insp 0 \
--if_hypothesis_suggstor 0 \
--api_key sk-