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

Using -model_file pt_inception5h.pth

Open Bird-NZ opened this issue 2 years ago • 0 comments

Hi, I'm building a model based on pt_inception5h.pth with the help of the arg "-model_file" but when I run this in "https://github.com/ProGamerGov/neural-dream/tree/dream-creator-support" I get:

`Successfully loaded Dream-Creator model: models/1model5h_v1120.pth
Setting up dream layer 1: inception_4d_3x3_reduce
Traceback (most recent call last):
  File "neural_dream.py", line 803, in <module>
    main()
  File "neural_dream.py", line 162, in main
    has_inception, layerList, params.classify, start_params, primary_params, secondary_params)
  File "D:\OneDrive\GitHub\neural-dream-dream-creator-support\neural_dream\dream_model.py", line 139, in build_net
    loss_module, sn = add_to_incept(net_base, n, sn, loss_module, module_loc)
  File "D:\OneDrive\GitHub\neural-dream-dream-creator-support\neural_dream\dream_model.py", line 14, in add_to_incept
    getattr(net, layer_name[0]).register_forward_hook(loss_module)
  File "C:\Users\OEM\Anaconda3\envs\tf-gpu\lib\site-packages\torch\nn\modules\module.py", line 772, in __getattr__
    type(self).__name__, name))
torch.nn.modules.module.ModuleAttributeError: 'InceptionV1_Caffe' object has no attribute 'inception_4d_3x3_reduce'`

Bird-NZ avatar Jun 01 '22 01:06 Bird-NZ