noise2noise
noise2noise copied to clipboard
models下的三个模型均不能使用
RuntimeError: Error(s) in loading state_dict for UNet:
Missing key(s) in state_dict: "_block2_2.0.weight", "_block2_2.0.bias", "_block4_1.0.weight", "_block4_1.0.bias", "_block4_1.2.weight", "_block4_1.2.bias", "_block4_1.4.weight", "_block4_1.4.bias".
size mismatch for _block4.0.weight: copying a param with shape torch.Size([96, 96, 3, 3]) from checkpoint, the shape in current model is torch.Size([96, 144, 3, 3]).
请问最后怎么解决的?
请问最后怎么解决的?
你好,请问你解决这个问题了吗