1. 背景
在深度學習的任務中,通常需要比較復雜的參數以及輸入輸出配置,比如需要不同的訓練data,不同的模型,寫入不同的log文件,輸出到不同的文件夾以免混淆輸出
常用的parser.add()方法非常占用代碼空間,而且輸入輸出配置無法通過文件更改,只能通過命令行參數改變。
docopt 庫提供了一個非常優雅的命令行解析工具,這邊只介紹其中解析文件參數
2. 安裝
pip install docopt
import docopt import json
3. 使用
docopt 由兩部分組成,第一部分是解析部分,通過“”“ -”“”(一段類似注釋的雙引號表示),這一部分必不可少。
""" Usage: train"""
上面解析部分的指令很簡單,輸入應為:
python test.py config.json
第二部分是main 函數:
if __name__ == "__main__": args = docopt.docopt(__doc__) args = json.load(open(args[""])) print('==>Params') for key in args.keys(): print('\t{}:{}\n'.format(key,args[key])) train(args)
我將 config.json 設置為:
{ "dataset": "human_science", "length": 1000, "model": "CNN", "log_dir": "./logs/", "output_dir": "./output/", "output_prefix": "human_science_CNN", "lr":0.01 }
命令行的輸出應為:
成功將參數傳入train()函數
以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支持腳本之家。
更多文章、技術交流、商務合作、聯系博主
微信掃碼或搜索:z360901061

微信掃一掃加我為好友
QQ號聯系: 360901061
您的支持是博主寫作最大的動力,如果您喜歡我的文章,感覺我的文章對您有幫助,請用微信掃描下面二維碼支持博主2元、5元、10元、20元等您想捐的金額吧,狠狠點擊下面給點支持吧,站長非常感激您!手機微信長按不能支付解決辦法:請將微信支付二維碼保存到相冊,切換到微信,然后點擊微信右上角掃一掃功能,選擇支付二維碼完成支付。
【本文對您有幫助就好】元
