GitLab

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

* Initial version of classification reference training script

* Updates

* Minor updates

* Expose a few more options

* Load optimizer and lr_scheduler when resuming

Also log the learning rate

* Evaluation-only and minor improvements

Identified a bug in the reporting of the results. They need to be reduced between all processes

* Address Soumith's comment

* Fix some approximations on the evaluation metric

* Flake8

* Add test for loading pretrained models

The update modifies the test to check whether the model can successfully load the pretrained weights. Will raise an error if the model parameters are incorrectly defined or named.

* Add test on 'num_class'

Passing num_class equal to a number other than 1000 helps in making the test more enforcing in nature.

* Added Semantic Boundaries Dataset

* Updated docs and datasets importing module

* Improved docs

* Add basic model testing.

Also fixes flaky test

* Fix flake8

* 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.

* Update transform doc with random offset of padding due to pad_if_needed

Updating transform docs with the random offset of padding when pad_if_needed is set to True.

* Update docs

Raise an `IndexError` for `FakeData.__getitem__()` if the index would be out of range.

* Added dimensions in the comments

The update provides the dimensions of the processed data following the style of inceptionV3 implementation.

* Changed docs and comments

Updated doc with the argument `transform_input`. Modified comments to match inceptionV3 style.

Including the `transform_input` argument in the docs of inceptionV3

* Add Caltech101 and Caltech256

* Add information about default for target_type

* Fix docs

* Add function to download from Google Drive

* Add CelebA dataset

* Only import pandas when needed

* Addressing comments

* Remove trailing whitespace

* Replace torch.LongTensor with torch.as_tensor

* Add GoogLeNet (Inception v1)

* Fix missing padding

* Add missing ReLu to aux classifier

* Add Batch normalized version of GoogLeNet

* Use ceil_mode instead of padding and initialize weights using "xavier"

* Match BVLC GoogLeNet zero initialization of classifier

* Small cleanup

* use adaptive avg pool

* adjust network to match TensorFlow

* Update url of pre-trained model and add classification results on ImageNet

* Bugfix that improves performance by 1 point

Remove explicit Python interpreter name.

* Fixes for PyTorch version of tqdm

* Flake

* Flake fix

* randomresizedmods

* lint checks

* test to randomrescrop added

* updates

* tests updated

* tests updated

* upd

* updates

* Update torchvision/transforms/transforms.py
Co-Authored-By: surgan12 <33121121+surgan12@users.noreply.github.com>

* tests changed

* trvis

* travis

* fixes syntax

* ...

* flake fixes

* flake_fixes

* flake_fixes2

* flake_fixes

* flake_fixes2