1234567891011121314151617181920 |
- import os
- model = 'meta-llama/Meta-Llama-3-8B-Instruct'
- names = [
- 'gsm8k-llama3-8b-new-mcts-8',
- 'gsmhard-llama3-8b-new-mcts-8',
- 'olympiadbench-llama3-8b-new-mcts-8',
- 'GAIC-llama3-8b-new-mcts-8',
- 'MATH-llama3-8b-new-mcts-8',
- 'AIME-llama3-8b-mcts-2'
- ]
- # DATA_NAME = 'gsmhard-llama3-8b-new-mcts-8'
- # DATA_NAME = 'olympiadbench-llama3-8b-new-mcts-8'
- # DATA_NAME = 'GAIC-llama3-8b-new-mcts-8'
- # DATA_NAME = 'MATH-llama3-8b-new-mcts-8'
- # DATA_NAME = 'AIME-llama3-8b-mcts-2']
- for name in names:
- os.system(f'sbatch -p vip_gpu_ailab -A ai4phys --gres=gpu:1 batch_run.sh {model} {name}')
|