解释代码: if args.model == 'dncnn': net = DnCNN(args.D, C=args.C).to(device) elif args.model == 'udncnn': net = UDnCNN(args.D, C=args.C).to(device) elif args.model == 'dudncnn': net = DUDnCNN(args.D, C=args.C).to(device) else: raise NameError('Please enter: dncnn, udncnn, or dudncnn')
时间: 2023-06-03 09:02:40 浏览: 178
DNN模型实现代码(带注释)
这段代码是一个用于选择模型的条件语句。如果 args.model 等于 'dncnn',则创建一个 DnCNN 模型并将其移动到指定设备上;如果 args.model 等于 'udncnn',则创建一个 UDnCNN 模型,并将其移动到指定设备上;如果 args.model 等于 'dudncnn',则创建一个 DUDnCNN 模型,并将其移动到指定设备上。如果 args.model 不等于这三种情况,则会输出一个 NameError 错误信息,提示用户输入正确的模型参数。
阅读全文