New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Example not working in Jupyter NBs; #95

Open

maximilianmandl opened this issue May 25, 2020 · 2 comments

Assignees

maximilianmandl commented May 25, 2020

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 self-assigned this

Contributor

ailzhang commented May 26, 2020

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

Contributor

ailzhang commented May 26, 2020

@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

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment