python库argparse中type的新奇指定方法

2024年4月15日 169次阅读 来源: Danno

最近在看一些项目的源码，总是能学到好多东西。

关于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.')
    原文作者：Danno
    原文地址: https://www.cnblogs.com/shiwanghualuo/p/11705204.html
    本文转自网络文章，转载此文章仅为分享知识，如有侵权，请联系博主进行删除。
点赞