Vision: doc-models.inception_v3: Add information about expected input size

Created on 28 Jul 2018  路  5Comments  路  Source: pytorch/vision

While using the pretrained inception v3 model I wasnt aware that the input size has to be 299x299, as I figured out after a little bit of try and error and searching.

I hope I didn't miss something in the docs, but if not it might make sense to add a small note in the models.inception_v3 doc that the expected tensor size is different for this model.

enhancement help wanted

All 5 comments

This is currently "documented" in https://github.com/pytorch/vision/blob/master/torchvision/models/inception.py#L77, but I agree that it would be better to have it more visible in the official documentations.
Could you send a PR improving the documentation?

Sure, I take that!

Sorry for the late response! had a bit of an issue with the doc building process to verify it looks properly, but a PR is in place now!

@fmassa I think this can be closed :)

I'll close it 馃憤

Was this page helpful?
0 / 5 - 0 ratings

Related issues

Abolfazl-Mehranian picture Abolfazl-Mehranian  路  3Comments

alpha-gradient picture alpha-gradient  路  3Comments

yxlabs picture yxlabs  路  4Comments

Linardos picture Linardos  路  4Comments

datumbox picture datumbox  路  3Comments