本文主要是介绍模型训练----parser.add_argument添加配置参数,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
现在需要配置参数来达到修改训练的方式,我现在需要新建一个参数来开关wandb的使用。
首先就是在def parse_option():函数里添加上你要使用的变量名
parser.add_argument("--open_wandb",type= bool,default=False,help='open wandb')
到config文件里增加你的变量
最后在到config后边将这两个变量的值连接起来
你就可以在程序你使用这个config.TRAIN.OPEN_WANDB变量了
点击访问博客查看更多内容 |
---|
这篇关于模型训练----parser.add_argument添加配置参数的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!