Python常用编程(Python-argparse库)

2021-03-07 20:29

阅读:682

标签:ntp   RKE   创建对象   参考   weight   imp   nap   esc   val   

参考:https://blog.csdn.net/foneone/article/details/103984895

 argparse库是一个存储参数库,可以用来进行模型训练过程中的参数保存作为一个整体,以便于使用和更改。

1、创建

import argparse
parser = argparse.ArgumentParser(description=Testing...) #创建对象

2、添加参数

# Input Parameters
    parser.add_argument(--orig_images_path, type=str,
                        default="data/clear/")
    parser.add_argument(--hazy_images_path, type=str, default="data/hazy/")
    parser.add_argument(--lr, type=float, default=0.0001)
    parser.add_argument(--weight_decay, type=float, default=0.0001)
    parser.add_argument(--grad_clip_norm, type=float, default=0.1)
    parser.add_argument(--num_epochs, type=int, default=10)
    parser.add_argument(--train_batch_size, type=int, default=8)
    parser.add_argument(--val_batch_size, type=int, default=8)
    parser.add_argument(--num_workers, type=int, default=4)
    parser.add_argument(--display_iter, type=int, default=10)
    parser.add_argument(--snapshot_iter, type=int, default=200)
    parser.add_argument(--snapshots_folder, type=str, default="snapshots/")
    parser.add_argument(--sample_output_folder, type=str, default="samples/")

    config = parser.parse_args()

 

Python常用编程(Python-argparse库)

标签:ntp   RKE   创建对象   参考   weight   imp   nap   esc   val   

原文地址:https://www.cnblogs.com/kyshan/p/14256276.html


评论


亲,登录后才可以留言!