读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
如果都多个函数的话: