github上下载的深度学习代码,为快速在jupyter notebook运行替换ArgumentParser定义的参数

🔥 🔥 🔥 github上的参数代码一般是如下格式:

if __name__ == '__main__':
    parser = ArgumentParser()
    parser.add_argument("--test_dir", type=str, default="test", help='Directory containing the test data (must have subdirectory noisy/)')
    parser.add_argument("--enhanced_dir", type=str, default="test", help='Directory containing the enhanced data')
    parser.add_argument("--ckpt", type=str, default="logs/epoch=326-step=408750.ckpt", help='Path to model checkpoint.')
    parser.add_argument("--corrector", type=str, choices=("ald", "langevin", "none"), default="ald", help="Corrector class for the PC sampler.")
    parser.add_argument("--corrector_steps", type=int, default=1, help="Number of corrector steps")
    parser.add_argument("--snr", type=float, default=0.5, help="SNR value for (annealed) Langevin dynmaics.")
    parser.add_argument("--N", type=int, default=30, help="Number of reverse steps")
    args = parser.parse_args()

👉👉👉 为了让jupyter notebook运行代码,需要将ArgumentParser定义的参数替换,但是又由于项目中用了很多args的属性,因此定义一个需要定义 一个 args.参数的函数,如下替换就可以:

    class Point_Attribute(dict):
    
        #给字典中加入点属性
        def __init__(self, **kwargs):
            super().__init__(kwargs)

        def __setattr__(self, key, value):
            #设置属性
            self[key] = value

        def __getattr__(self, key):  #当出现iris.datas时,调用getattr方法
            #获取属性
            self[key]
            return self[key]  

    args = Point_Attribute(
        test_dir="test",
        enhanced_dir="/kaggle/working/", 
        ckpt="logs/epoch=326-step=408750.ckpt", 
        corrector="ald", 
        corrector_steps=1, 
        snr=0.5,
        N=4
    )
  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值