Example not working in Jupyter NBs;

[maximilianmandl]

Hi there,

I wanted to play around with your SSD300 - thanks again for creating that, btw - as it is probably better than the implementation I've been using. However, just trying to get the easy examples to run in Jupyter Notebooks has been a struggle. I used conda to install all the packages and correct versions as posted but when I try to predict the sample images, I get an error (screenshot):

Full code, copy-pasted from https://pytorch.org/hub/nvidia_deeplearningexamples_ssd/

Any ideas if this is a bug or something I need to look into deeper on my end?

Thanks again for putting out great stuff!

[ailzhang]

Thanks for the report! I'll take a look there soon!

[ailzhang]

@maximilianmandl What's your pytorch version btw? I've tried it on Colab and it works as expected. https://colab.research.google.com/github/pytorch/pytorch.github.io/blob/master/assets/hub/nvidia_deeplearningexamples_ssd.ipynb#scrollTo=H8caObhcgPqK