python库argparse中type的新奇指定方法
程序员文章站
2024-01-23 11:30:28
最近在看一些项目的源码,总是能学到好多东西。 关于arparse中type的类型指定 不止可以指定常规类型,还可以加一些自己类型判断,具体用法如下( "来源" ): ......
最近在看一些项目的源码,总是能学到好多东西。
关于arparse中type的类型指定
不止可以指定常规类型,还可以加一些自己类型判断,具体用法如下(来源):
def str2bool(v): """ usage: parser.add_argument('--pretrained', type=str2bool, nargs='?', const=true, dest='pretrained', help='whether to use pretrained models.') """ if v.lower() in ('yes', 'true', 't', 'y', '1'): return true elif v.lower() in ('no', 'false', 'f', 'n', '0'): return false else: raise argparse.argumenttypeerror('unsupported value encountered.') if __name__ == "__main__": parser = argparse.argumentparser() parser.add_argument('--syncbn', type=str2bool, nargs='?', default=false, dest='network.syncbn', help='whether to sync bn.')
上一篇: Python之数据类型转换
下一篇: RabbitMQ安装