stevehuanghe / image_captioning

Image captioning models in PyTorch
Apache License 2.0
37 stars 10 forks source link

Problem in testing SSA #9

Open glados97 opened 3 years ago

glados97 commented 3 years ago

Hello, im trying to get the results from the test.py using SSA model but i encounter this error:

 File "test.py", line 150, in <module>
    main()
  File "test.py", line 116, in main
    results = net.beam_search(images, start_token, beam_width=args.beam_width).data.cpu().numpy()
  File "/home/----/image_captioning/models/ssa.py", line 142, in beam_search
    features = self.bn(self.linear(features))
  File "/home/---/anaconda3/lib/python3.7/site-packages/torch/nn/modules/module.py", line 772, in __getattr__
    type(self).__name__, name))
torch.nn.modules.module.ModuleAttributeError: 'SSA' object has no attribute 'bn'

If you suspect this is an IPython 7.18.1 bug, please report it at:
    https://github.com/ipython/ipython/issues
or send an email to the mailing list at ipython-dev@python.org

You can print a more detailed traceback right now with "%tb", or use "%debug"
to interactively debug it.

Extra-detailed tracebacks for bug-reporting purposes can be enabled via:
    %config Application.verbose_crash=True