diff options
| author | d8ahazard <d8ahazard@gmail.com> | 2022-09-30 12:51:54 -0500 |
|---|---|---|
| committer | d8ahazard <d8ahazard@gmail.com> | 2022-09-30 12:51:54 -0500 |
| commit | 8f1d412e7bcd279c3c49b0153def0e3d6d941e5a (patch) | |
| tree | b7f35abcabb92ed4cb7cb90537f3a89be858fa53 /modules/bsrgan_model.py | |
| parent | 9fc1e49bd2674f8c30dd3545c1c271472ff6d3c2 (diff) | |
Fix BSRGAN Model loading.
Diffstat (limited to 'modules/bsrgan_model.py')
| -rw-r--r-- | modules/bsrgan_model.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/modules/bsrgan_model.py b/modules/bsrgan_model.py index 339f402b..165f77b5 100644 --- a/modules/bsrgan_model.py +++ b/modules/bsrgan_model.py @@ -69,7 +69,8 @@ class UpscalerBSRGAN(modules.upscaler.Upscaler): if not os.path.exists(filename) or filename is None: print("Unable to load %s from %s" % (self.model_path, filename)) return None - model = RRDBNet(in_nc=3, out_nc=3, nf=64, nb=23, gc=32, sf=2) # define network + print("Loading %s from %s" % (self.model_path, filename)) + model = RRDBNet(in_nc=3, out_nc=3, nf=64, nb=23, gc=32, sf=4) # define network model.load_state_dict(torch.load(filename), strict=True) model.eval() for k, v in model.named_parameters(): |