-
Notifications
You must be signed in to change notification settings - Fork 225
Description
is there any way to way to load a metrics weights, downloaded to some other path in the system.
i loaded the lpips with no weights. then loaded from a custumj path but its showing me below error.
model = pyiqa.create_metric('lpips',pretrained=False)
model.net.load_state_dict(torch.load('/media/abdulsami/New Volume/LPIPS_v0.1_alex-df73285e.pth')['params'])
KeyError: 'params'
also without params
RuntimeError: Error(s) in loading state_dict for LPIPS:
Missing key(s) in state_dict: "scaling_layer.shift", "scaling_layer.scale", "net.slice1.0.weight", "net.slice1.0.bias", "net.slice2.3.weight", "net.slice2.3.bias", "net.slice3.6.weight", "net.slice3.6.bias", "net.slice4.8.weight", "net.slice4.8.bias", "net.slice5.10.weight", "net.slice5.10.bias", "lins.0.model.1.weight", "lins.1.model.1.weight", "lins.2.model.1.weight", "lins.3.model.1.weight", "lins.4.model.1.weight".