读github上的很多pytorch源码时,发现大部分用的都是argparse来进行参数解析,虽然功能强大,但是还是复杂了一些,后来发现有谷歌开源fire库,发现参数解析变得简单了很多呢~
首先,安装fire:
pip install fire
超简单使用范例:
# save the file as train.py
import fire
def train(p1=0, p2=1, p3=2):
print(p1, p2, p3)
if __name__ == '__main__':
fire.Fire(train)
# eg: python train.py --p1=2 --p2=0 --p3=1
如果都多个函数的话:

本文介绍了谷歌开源的Python参数解析库Fire,它使得参数解析变得更简单。通过示例展示了如何安装和基本使用Fire,包括调用多个函数和查看帮助。尽管在参数注释方面稍显不足,但对于日常调参任务而言,Fire是一个实用的工具。
最低0.47元/天 解锁文章
1692

被折叠的 条评论
为什么被折叠?



