深度學(xué)習(xí)代碼中argparse以及yaml文件的使用
點(diǎn)擊上方“程序員大白”，選擇“星標(biāo)”公眾號(hào)
重磅干貨，第一時(shí)間送達(dá)
01
if?__name__?==?'__main__':parse = argparse.ArgumentParser()parse.add_argument('--num_batches', type=int, default=50, help='the num of batch')parse.add_argument('--num_window', type=int, default=5, help='the num of window')parse.add_argument('--weight', type=str, default= '../pretrain.pth', help='the path of pretrained model')opt = parse.parse_args()
import argparsedef main(opt):print(opt.num_batches)if __name__ == '__main__':parse = argparse.ArgumentParser()parse.add_argument('--num_batches', type=int, default=50, help='the num of batch')parse.add_argument('--num_window', type=int, default=5, help='the num of window')parse.add_argument('--weight', type=str, default= '../pretrain.pth', help='the path of pretrained model')opt = parse.parse_args()????main(opt)
02
device: 'cpu'data:train_path: 'data/train'test_path: 'test/train'????num:?1000
03
def read_yaml(path):file = open(path, 'r', encoding='utf-8')string = file.read()dict = yaml.safe_load(string)return dict
path = 'config.yaml'Dict = read_yaml(path)device = Dict['device']print(device)train_path = Dict['data']['train_path']print(train_path)
import yamldef read_yaml(path):file = open(path, 'r', encoding='utf-8')string = file.read()dict = yaml.safe_load(string)return dictpath = 'config.yaml'Dict = read_yaml(path)device = Dict['device']print(device)train_path = Dict['data']['train_path']print(train_path)
04
推薦閱讀
關(guān)于程序員大白
程序員大白是一群哈工大，東北大學(xué)，西湖大學(xué)和上海交通大學(xué)的碩士博士運(yùn)營(yíng)維護(hù)的號(hào)，大家樂(lè)于分享高質(zhì)量文章，喜歡總結(jié)知識(shí)，歡迎關(guān)注[程序員大白]，大家一起學(xué)習(xí)進(jìn)步！
評(píng)論
圖片
表情