-
Notifications
You must be signed in to change notification settings - Fork 34
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.
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
training result #9
Comments
And I transform the final checkpoint_60.ckpt to checkpoint_60.pth only using the "model" parameter in checkpoint_60.ckpt,but it reports errors when testing :" RuntimeError: Error(s) in loading state_dict for Harmonizer: Missing key(s) in state_dict: "backbone._blocks.0._depthwise_conv.weight",..." Thanks ahead for your help! import torch
# 指定 checkpoint 文件的路径
checkpoint_path = 'checkpoint_60.ckpt'
checkpoint = torch.load(checkpoint_path)
# 提取模型的参数
model_state_dict = checkpoint['model']
torch.save(model_state_dict, "checkpoint_60.pth") |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Hi, I used the training code you provided for training, but didn't find the result of the training which is a similar file like the harmonizer.pth you provided. May I know where this file is stored after training? Or what part of the code needs to be changed to get this file?
The text was updated successfully, but these errors were encountered: