Skip to content

Using -model_file pt_inception5h.pth #17

@Bird-NZ

Description

@Bird-NZ

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'`

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions