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

Dreaming over dream-creator fine-tuned model #14

Open
flaclad opened this issue Jun 26, 2021 · 1 comment
Open

Dreaming over dream-creator fine-tuned model #14

flaclad opened this issue Jun 26, 2021 · 1 comment

Comments

@flaclad
Copy link

flaclad commented Jun 26, 2021

Hi,
I would like to dream over a model created by finetuning pretrained BVLC with your other repo 'dream-creator' but I get this error "ValueError: Model architecture not recognized."
Any quick advise about how to solve it?
By the way these repoes are amazing

@Bird-NZ
Copy link

Bird-NZ commented Jul 11, 2022

Hi @flaclad Did you ever get to the bottom of this?

I'm getting similar issues with any model I build from pt_inception5h.pth over in dream-creator https://github.com/ProGamerGov/dream-creator and try and then try to run in the "neural-dream-dream-creator-support" Branch

Error detail I get:
RuntimeError: Error(s) in loading state_dict for Inception5h:

Missing key(s) in state_dict: "conv2d0_pre_relu_conv.weight", "conv2d0_pre_relu_conv.bias", "conv2d1_pre_relu_conv.weight", "conv2d1_pre_relu_conv.bias", "conv2d2_pre_relu_conv.weight", "conv2d2_pre_relu_conv.bias",.................
................
Unexpected key(s) in state_dict: "model_state_dict", "epoch", "optimizer_state_dict", "lrscheduler_state_dict", "normalize_params", "num_classes", "has_branches", "base_model".

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