GitLab

* final changes

* final

* linter

* test changes

* linter

* lint

* indent

* lint

* minor changes

* parameter added

* ci

* ci fixes

* indent

* indent

* indent

* arg fixed

* fixed check integrity

* stylistic changes

* flake8

* added is_valid_file option

* small fixes

* fixes

* flake8 fixes

* some test

* flake8 fixes

* improvements

* modifications on tests
* fixes

* minor fix

* Add dilation option to ResNet

* Add a size check for replace_stride_with_dilation

* Fix wrong doc string

* bug fix

* Fix ResNeXt model defs with backwards compat for ResNet.

* Fix Python 2.x integer div issue

* fix inconsistent behavior for ~ expression

* remove duplicated code

* Update test_transforms.py

* Update transforms.py

* introduced new super class for all vision datasets

* Removed root from repr if dataset has no root

* reverted some overly-ambitious autoformatting

* reverted some overly-ambitious autoformatting

* added split attribute to repr of STL10 dataset

* made Python2 friendly and more robust

* Fixed call of the superclass constructor

* moved transform and target_transform back to the base classes

* added check if transforms are present before printing to avoid setting them within the constructor

* added missing transforms and target_transforms to base classes

* fixed linter error

* 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

str.endswith can take a single string, or a tuple of strings. This replaces a loop over each extension and a call to endswith with a single call to endswith passing in all the allowed extensions. This has the advantage that the loop over each extension is done in c rather than python, and the code is a little less verbose.

* Add groups support to ResNet

* Kill BaseResNet

* Make it support multi-machine training

Making references/classification/train.py and references/classification/utils.py  compatible with python2  (#831)

* linter fixes

* linter fixes

* remove duplicate code from densenet

* correct indentation

* Match Tensorflows implementation of GoogLeNet

* just disable the branch when pretrained is true

* don't use legacy code

Changed 'channely' to 'channel' in the comments of normalize function.

* Add MobileNet V2

* Remove redundant functions and make tests pass

* Simplify a bit the implementation

* Reuse ConvBNReLU more often

* Remove input_size and minor changes

* Py2 fix

* Added Semantic Boundaries Dataset

* Updated docs and datasets importing module

* Improved docs

* Add Affinetransformation

Add Affinetransformation to superseed LinearTransformation

* Add test

* Add zero mean_vector in LinearTransformation and improved docs

* update

* minor fix

* minor fix2

* fixed flake8

* fix flake8

* fixed transpose syntax

* fixed shape of mean_vector in test

* fixed test

* print est cov and mean

* fixed flake8

* debug

* reduce num_samples

* debug

* fixed num_features

* fixed rtol for cov

* fix __repr__

* Update transforms.py

* Update test_transforms.py

* Update transforms.py

* fix flake8

* Update transforms.py

* Update transforms.py

* Update transforms.py

* Update transforms.py

* Changed dim of mean_vector to 1D, doc and removed .numpy () from format_string

* Restore test_linear_transformation()

* Update test_transforms.py

Signed-off-by: Michal Zientkiewicz <michalz@nvidia.com>

* initial commit

* fixed Python2 issue

* fixed naming incorrectness and Python2 compability

* fixed preparation of train folder

* removed detection dataset

* added docstring and repr

* moved import of scipy to make the import of torchvision independent of it

* improved conversion from class string to index

* removed support for other years than 2012

* removed accidentally added file

* moved emptying of split folder to avoid accidental deletion

* removed deletion of the images

* removed error conversion for Python2

* Aligned class indices with the indices identified by ImageFolder class

In reference to #729 added comments to clarify the naming and action of the layers performing downsampling in resnets.