arg README
这是一个用于debug参数解析的插件,是否还在为了手写一长串的参数而烦恼。
比如如下一个训练参数,当你想debug一下时
python trainer_rl.py --configs defaults_rlhf --cache_dir $DATA_PATH --rank_model $REWARD_MODEL --sft_model $SFT_MODEL --output_dir $MODEL_PATH/rl_model
你不得不手动的在vscode的args里输入一个列表形式的参数,很麻烦不是吗。
Features
这个插件实现的就是把参数解析成数组(很简单的功能)
步骤
- 复制参数列表即上述的
--configs defaults_rlhf --cache_dir $DATA_PATH --rank_model $REWARD_MODEL --sft_model $SFT_MODEL --output_dir $MODEL_PATH/rl_model 注意 上面 python trainer_rl.py不应该复制下来因为launch.json这个参数会写在别的地方
- 键盘
ctrl+shift+p,然后输入split arg然后把参数粘贴到输入框中即可
- 分割的结果会复制到你的剪切板,你只需要粘贴到args里就好了
| |