New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

problem in weights/600.pth #1

Open

Seyed-Ali-Ahmadi opened this issue Feb 22, 2022 · 1 comment

Seyed-Ali-Ahmadi commented Feb 22, 2022

Hi,

I'm trying to reimplement your code, but when I want to do the following piece of code, I get an error:

weight = torch.load(ckpt)
model = Unet(1, 8).cuda().eval()
model.load_state_dict(weight)

Error:

RuntimeError: Error(s) in loading state_dict for Unet:
	Missing key(s) in state_dict: "conv_up1.conv_up1.weight", "conv_up1.conv_up1.bias", "conv_up2.conv_up1.weight", "conv_up2.conv_up1.bias". 
	Unexpected key(s) in state_dict: "conv_up1.weight", "conv_up1.bias", "conv_up2.weight", "conv_up2.bias".

What should I do?

The text was updated successfully, but these errors were encountered:

Contributor

Qian2333 commented Mar 8, 2022

sorry, I checked the weight and found I uploaded the wrong file, the problem will be fixed soon.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment