dk008652/dst_grpo

★ 0Forks 0PythonGitHub ↗Compare

README

#DST_GRPO

1. Environment

conda env create -f environment.yml conda activate dst-grpo

2. Configure token

edit config.json → set your token

3. Dataset

unzip mwz2.4.zip python prepare_grpo_dataset.py --out grpo_dataset

4. Train

python train_dst_grpo.py
--out llama_grpo_v1
--checkpoint llama_AMD_value_0.10/checkpoint-13696
--max_steps 2000 --eval_steps 200 --save_steps 200 \

output_grpo_v1.log 2>&1

5. Generate

python generate_grpo.py
--model_dir llama_grpo_v1
--adapter llama_grpo_v1/checkpoint-2000

Contributors

dk008652

Issues