neural-dream
neural-dream copied to clipboard
Dreaming over dream-creator fine-tuned model
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
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".