标签: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()
标签:ntp RKE 创建对象 参考 weight imp nap esc val
原文地址:https://www.cnblogs.com/kyshan/p/14256276.html