GitLab

Co-authored-by: Nicolas Hug <nicolashug@fb.com>

Summary: We would like to track the internal usage of TorchVision components. We're tracking [datasets usage](https://fburl.com/daiquery/dqpmemn3

) now. This diff expand the tracking to all models.

Reviewed By: fmassa

Differential Revision: D31441632

fbshipit-source-id: e26072e582ac9f832c2056307ebf0eccf2ed6c9c
Co-authored-by: Kai Zhang <kaizh@fb.com>

* Added Dropout parameter of Models

* Added argument description for dropout in MobileNet v2 and v3
Updated quantization/googlenet.py as per the changes in constructor in googlenet

* Moved new dropout parameter n the end

* Updated googlenet.py
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>

* add ufmt as code formatter

* cleanup

* quote ufmt requirement

* split imports into more groups

* regenerate circleci config

* fix CI

* clarify local testing utils section

* use ufmt pre-commit hook

* split relative imports into local category

* Revert "split relative imports into local category"

This reverts commit f2e224cde2008c56c9347c1f69746d39065cdd51.

* pin black and usort dependencies

* fix local test utils detection

* fix ufmt rev

* add reference utils to local category

* fix usort config

* remove custom categories sorting

* Run pre-commit without fixing flake8

* got a double import in merge
Co-authored-by: Nicolas Hug <nicolashug@fb.com>

using nn.init.trunc_normal_ instead of scipy.stats.truncnorm
Co-authored-by: Vincent Moens <vmoens@fb.com>

* [doc] add minimum input size for alexnet builder

* [doc] add minimum input size for vgg builder

* [doc] add minimum input size for squeezenet builder

* [doc] add minimum input size for densenet builder

* [doc] add minimum input size for inception_v3 builder

* [doc] add minimum input size for googlenet builder
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>

* Replacing all torch.jit.annotations with typing

* Replacing remaining typing

* Making quantized inception torchscriptable.

* Adding a test.

* Fix mypy warning.

* style: Added annotation typing for inception

* refactor: Moved factory function after class definition

* style: Changed attribute setting for type hinting

* refactor: Removed un-necessary import

* fix: Fixed typing in constructors

* fix: Fixed kwargs typing

* style: Fixed lint

* refactor: Moved helpers function back and quote typed it

* partially enable mypy for .models

* fix existing errors

* ignore error instead of using Union

* remove sys.version_info == 2

* remove sys.version_info < 3

* remove from __future__ imports

* quantizable googlenet

* Minor improvements

* Rename basic_conv2d with conv_block plus additional fixes

* More renamings and fixes

* Bugfix

* Fix missing import for mypy

* Add pretrained weights

* make googlnet scriptable

* Remove typing import in favor of torch.jit.annotations

* add inceptionnet

* flake fixes

* fix asssert true

* add import division for torchscript

* fix script compilation

* fix flake, py2 division error

* fix py2 division error

Fix the old flatten method which use the size(0) to caculate the batch size, the old method will intruduce Gather opertion in the onnx output, which will faild parsed by tensorRT 5.0 (#1134)

I grepped the repo for Ouputs and these were the only occurences

* added progress flag to model getters

* flake8

* bug fix

* backward commpability

* add aux_logits support to inception

it is related to pytorch/pytorch#18668

* instantiate InceptionAux only when requested

it is related to pytorch/pytorch#18668

* revert googlenet

* support and aux_logits in pretrained models

* return namedtuple when aux_logit is True

Including the `transform_input` argument in the docs of inceptionV3

* flake_fixes

* flake_fixes2

* Modifying the comments of inceptionV3 dimensions

Modifying the comments of inceptionV3 dimensions to match the pytorch convention. Relevant (https://github.com/pytorch/vision/pull/719#pullrequestreview-203194302)

* Added Batch size in comment

* Update inception.py

* Updated inceptionV3 to accept different sized images (Adaptive avg pool)

The update allows inceptionV3 to process images larger or smaller than prescribed image size (299x299) using adaptive average pooling. Will be useful while finetuning or testing on different resolution images.

* Update inception.py

* Added size information for inception v3

Since inception v3 expects in contrast to the other models a size
of 299x299x3 I added a small note in the docstring to point that out
(addresses #560)

* inception size info in note box; fixed line length

* removed trailing whitespace

* Fix inception v3 input transform for trace & onnx

* Input transform are in-place updates, which produce issues for tracing
and exporting to onnx.

* nit

* fix for PyTorch v0.4 (remove .data and add _ for the initializations in nn.init)

* fix m.**.**() style to nn.init.**(**) style

* remove .idea

* fix lines and indents

* fix lines and indents

* change to use `kaming_normal_`

* add `.data` for safety

* add nonlinearity='relu' for sure

* fix indents

First dimension is batch size, channel is the second