mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2026-01-15 14:47:16 +00:00
9 lines
357 B
Bash
Executable File
9 lines
357 B
Bash
Executable File
set_n_least_used_CUDA_VISIBLE_DEVICES 1
|
|
|
|
python train_reward_model.py --pretrain 'microsoft/deberta-v3-large' \
|
|
--model 'deberta' \
|
|
--strategy naive \
|
|
--loss_fn 'log_exp'\
|
|
--save_path 'rmstatic.pt' \
|
|
--test True
|