读github上的很多pytorch源码时,发现大部分用的都是argparse来进行参数解析,虽然功能强大,但是还是复杂了一些,后来发现有谷歌开源fire库,发现参数解析变得简单了很多呢~
首先,安装fire:
pip install fire
超简单使用范例:
# save the file as train.py
import firedef 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
如果都多个函数的话:
import fire
def f1(p1=1,p2=2):print('this is f1')print(p1,p2)def f2(p1=0, p2=1, p3=2):print('this is f2')print(p1, p2, p3)if __name__ == '__main__':fire.Fire()# eg: python train.py f1 --p1=0 --p2=1
# python train.py f2 --p1=2 --p2=0
忘记参数时查看帮助:
python train.py f1 -- --help
总的来说使用起来十分方便,唯一地缺点就是对于每个参数作用的注释不够吧,但拿来调调参啥的完全够了~
更多功能参考:
本文发布于:2024-01-28 07:10:26,感谢您对本站的认可!
本文链接:https://www.4u4v.net/it/17063970325693.html
版权声明:本站内容均来自互联网,仅供演示用,请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系,我们将在24小时内删除。
留言与评论(共有 0 条评论) |