Skip to content

Commit cf3bcc8

Browse files
authored
Update main_test_rrdb.py
1 parent ffaf92f commit cf3bcc8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

main_test_rrdb.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -102,7 +102,7 @@ def main():
102102

103103
from models.network_rrdb import RRDB as net
104104
model = net(in_nc=n_channels, out_nc=n_channels, nc=64, nb=23, gc=32, upscale=4, act_mode='L', upsample_mode='upconv')
105-
model.load_state_dict(torch.load(model_path), strict=False)
105+
model.load_state_dict(torch.load(model_path), strict=True) # strict=False
106106
model.eval()
107107
for k, v in model.named_parameters():
108108
v.requires_grad = False

0 commit comments

Comments
 (0)