提交 ac21d308 编写于 作者: A AUTOMATIC

fix for one error with BSRGAN, though it fails to work anyway #1109

上级 43c87ef0
......@@ -67,9 +67,8 @@ class UpscalerBSRGAN(modules.upscaler.Upscaler):
else:
filename = path
if not os.path.exists(filename) or filename is None:
print("Unable to load %s from %s" % (self.model_dir, filename))
print(f"BSRGAN: Unable to load model from {filename}", file=sys.stderr)
return None
print("Loading %s from %s" % (self.model_dir, filename))
model = RRDBNet(in_nc=3, out_nc=3, nf=64, nb=23, gc=32, sf=2) # define network
model.load_state_dict(torch.load(filename), strict=True)
model.eval()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册