search

PRIS-CV / AP-CNN_Pytorch-master

Weakly Supervised Attention Pyramid Convolutional Neural Network for Fine-Grained Visual Classification (TIP2021)
MIT License
47 stars 4 forks source link

cannot unsqueeze empty tensor #2

Closed PabloZheng closed 3 years ago

PabloZheng commented 3 years ago

excellent work! I have some difficulties when I tried to use AP-CNN to train my customized dataset. the error was raised in "line 452, def get_att_roi" , I wonder if this error has been encountered before and what I can do to avoid this

PabloZheng commented 3 years ago

excellent work! I have some difficulties when I tried to use AP-CNN to train my customized dataset. the error was raised in "line 452, def get_att_roi" , I wonder if this error has been encountered before and what I can do to avoid this

joanbu2 commented 3 years ago

你好，我也遇到相同的错误，请问你有什么办法避免这种情况吗？