neural-dream icon indicating copy to clipboard operation
neural-dream copied to clipboard

Dreaming over dream-creator fine-tuned model

Open flaclad opened this issue 3 years ago • 1 comments

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

flaclad avatar Jun 26 '21 19:06 flaclad

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".

Bird-NZ avatar Jul 11 '22 22:07 Bird-NZ