- 07 May, 2019 3 commits
-
-
ekka authored
* Minor refactoring of ShuffleNetV2 Added progress flag following #875. Further the following refactoring was also done: 1) added `version` argument in shufflenetv2 method and removed the operations for converting the `width_mult` arg to float and string. 2) removed `num_classes` argument and **kwargs from functions except `ShuffleNetV2` * removed `version` arg * Update shufflenetv2.py * Removed the try except block * Update shufflenetv2.py * Changed version from float to str * Replace `width_mult` with `stages_out_channels` Removes the need of `_getStages` function.
-
Francisco Massa authored
-
bddppq authored
-
- 06 May, 2019 2 commits
-
-
Adam J. Stewart authored
-
ekka authored
* remove 'input_size' parameter from shufflenetv2 * Update shufflenetv2.py
-
- 03 May, 2019 1 commit
-
-
Vitor Finotti Ferreira authored
-
- 02 May, 2019 1 commit
-
-
ekka authored
-
- 01 May, 2019 1 commit
-
-
Surgan Jandial authored
-
- 30 Apr, 2019 3 commits
-
-
Bar authored
* Add ShuffleNet v2 Added 4 configurations: x0.5, x1, x1.5, x2 Add 2 pretrained models: x0.5, x1 * fix lint * Change globalpool to torch.mean() call
-
Philip Meier authored
* added progress flag to model getters * flake8 * bug fix * backward commpability
-
Philip Meier authored
-
- 29 Apr, 2019 1 commit
-
-
bbowles-personal authored
-
- 26 Apr, 2019 1 commit
-
-
Francisco Massa authored
* [RFC] Add support for joint transformations in VisionDataset * Add joints transforms for VOC and SBD Breaking change in SBD, the xy_transform has been renamed transforms. I think this is fine given that we have not released a version of torchvision that contains it
-
- 25 Apr, 2019 7 commits
-
-
Surgan Jandial authored
-
Philip Meier authored
* fixed check integrity * stylistic changes * added test for check_md5 and check_integrity * flake8 * fix path to test file if not executed from test folder
-
Surgan Jandial authored
* final changes * final * linter * test changes * linter * lint * indent * lint * minor changes * parameter added * ci * ci fixes * indent * indent * indent * arg fixed
-
Philip Meier authored
* fixed check integrity * stylistic changes * flake8
-
Dhananjay authored
* added is_valid_file option * small fixes * fixes * flake8 fixes * some test * flake8 fixes * improvements * modifications on tests * fixes * minor fix
-
Philip Meier authored
-
Francisco Massa authored
-
- 24 Apr, 2019 2 commits
-
-
Francisco Massa authored
* Add dilation option to ResNet * Add a size check for replace_stride_with_dilation
-
Francisco Massa authored
-
- 18 Apr, 2019 2 commits
-
-
Philip Meier authored
* Fix wrong doc string * bug fix
-
Philip Meier authored
-
- 17 Apr, 2019 1 commit
-
-
Adam J. Stewart authored
-
- 16 Apr, 2019 1 commit
-
-
Ailing authored
* add all available models * simplify vision hubconf
-
- 15 Apr, 2019 1 commit
-
-
Ross Wightman authored
* Fix ResNeXt model defs with backwards compat for ResNet. * Fix Python 2.x integer div issue
-
- 12 Apr, 2019 2 commits
-
-
cormoran authored
* fix inconsistent behavior for ~ expression * remove duplicated code
-
Philip Meier authored
-
- 10 Apr, 2019 1 commit
-
-
Ashok M authored
-
- 09 Apr, 2019 5 commits
-
-
Ashok M authored
-
ekka authored
* Update test_transforms.py * Update transforms.py
-
Soumith Chintala authored
-
Soumith Chintala authored
-
Soumith Chintala authored
-
- 08 Apr, 2019 2 commits
-
-
Ashok M authored
-
Allan Wang authored
-
- 05 Apr, 2019 2 commits
-
-
Francisco Massa authored
-
Ashok M authored
-
- 04 Apr, 2019 1 commit
-
-
Philip Meier authored
* 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
-