Skip to content
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

pretrained model using #5

Open
ansleepykid opened this issue Feb 23, 2021 · 1 comment
Open

pretrained model using #5

ansleepykid opened this issue Feb 23, 2021 · 1 comment

Comments

@ansleepykid
Copy link

how can i use a pretrained model at first? your model is different from normal resnet networks, do you train any model as pretrained model and how can I git it

@y0-0n
Copy link

y0-0n commented Jun 23, 2022

I think your error is related with this line.

model.load_state_dict(new_state_dict)

I changed the above line to below.

model.load_state_dict(new_state_dict, strict=False)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants