Skip to content
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.

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

Inception v3 online docs input explenation #189

Open
Mumbaso21 opened this issue Mar 4, 2021 · 1 comment
Open

Inception v3 online docs input explenation #189

Mumbaso21 opened this issue Mar 4, 2021 · 1 comment

Comments

@Mumbaso21
Copy link

Mumbaso21 commented Mar 4, 2021

Hi,
this is not a real issue, but just a note regarding the online docs of the Inception_v3 at this link.
I think it is better to insert a note in the page about the input shape as stated in the code documentation as:

.. note::
        **Important**: In contrast to the other models the inception_v3 expects tensors with a size of
        N x 3 x 299 x 299, so ensure your images are sized accordingly.

N must be greater than 1, otherwise it will throw an error, the strange thing is that it is reported as a training error even in pretrained mode (pretrained=True).

In the online docs it is not reported so well.

Best regards.

@parmeet
Copy link
Contributor

parmeet commented Jul 22, 2021

During training, N needs to be greater than 1 because the model expects more than 1 value per channel. If you run the model in eval mode (model.eval()), it is OK to pass single image i.e N=1.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants