- 26 Mar, 2019 5 commits
-
-
ekka authored
-
ekka authored
* 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.
-
Francisco Massa authored
-
vfdev authored
* Added Semantic Boundaries Dataset * Updated docs and datasets importing module * Improved docs
-
ekka authored
-
- 25 Mar, 2019 2 commits
-
-
Francisco Massa authored
* Add basic model testing. Also fixes flaky test * Fix flake8
-
ekka authored
* 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
-
- 24 Mar, 2019 1 commit
-
-
ekka authored
-
- 20 Mar, 2019 1 commit
-
-
Michał Zientkiewicz authored
Signed-off-by:Michal Zientkiewicz <michalz@nvidia.com>
-
- 19 Mar, 2019 1 commit
-
-
Philip Meier authored
* 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
-
- 11 Mar, 2019 3 commits
-
-
ekka authored
In reference to #729 added comments to clarify the naming and action of the layers performing downsampling in resnets.
-
ekka authored
-
ekka authored
* 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
-
- 09 Mar, 2019 4 commits
-
-
Will Frey authored
Raise an `IndexError` for `FakeData.__getitem__()` if the index would be out of range.
-
ekka authored
* 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.
-
ekka authored
Including the `transform_input` argument in the docs of inceptionV3
-
Irvin Ho authored
-
- 07 Mar, 2019 3 commits
-
-
Bryan He authored
* 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
-
maria8899 authored
-
Michael Kösel authored
* 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
-
- 06 Mar, 2019 1 commit
-
-
Olivier Courtin authored
Remove explicit Python interpreter name.
-
- 01 Mar, 2019 1 commit
-
-
Francisco Massa authored
* Fixes for PyTorch version of tqdm * Flake * Flake fix
-
- 28 Feb, 2019 2 commits
-
-
Francisco Massa authored
-
Surgan Jandial authored
-
- 27 Feb, 2019 1 commit
-
-
Soumith Chintala authored
-
- 26 Feb, 2019 1 commit
-
-
Surgan Jandial authored
-
- 23 Feb, 2019 1 commit
-
-
Vishwak Srinivasan authored
-
- 21 Feb, 2019 1 commit
-
-
Tongzhou Wang authored
-
- 18 Feb, 2019 2 commits
-
-
surgan12 authored
* 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
-
surgan12 authored
* flake_fixes * flake_fixes2
-
- 14 Feb, 2019 4 commits
-
-
ekka authored
* 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
-
ekka authored
* 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
-
ekka authored
The update allows VGG to process images larger or smaller than prescribed imagenet size using adaptive average pooling. Will be useful while finetuning or testing on different resolution images. Similar to https://github.com/pytorch/vision/pull/643 and https://github.com/pytorch/vision/pull/672. I did not include adaptive avg pool in features or classifier block so that these predefined blocks can be used as it is.
-
ekka authored
The update allows Alexnet to process images larger or smaller than prescribed imagenet size using adaptive average pooling. Will be useful while finetuning or testing on different resolution images. Similar to https://github.com/pytorch/vision/pull/643 and https://github.com/pytorch/vision/pull/672. I did not include adaptive avg pool in features or classifier block so that these predefined blocks can be used as it is.
-
- 13 Feb, 2019 4 commits
-
-
Francisco Massa authored
The tests were previously taking 2 minutes, not they take 4 seconds
-
Francisco Massa authored
-
Alex Alemi authored
* Update URL for EMNIST Update the dataset and download URL for EMNIST to the url from the updated arxiv version. * Update mnist.py
-
Jonas authored
* 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
-
- 12 Feb, 2019 2 commits
-
-
Michael Kösel authored
-
Michael Kösel authored
-