- 16 Nov, 2020 1 commit
-
-
Milos authored
* Fix MNASNet docstrings so they are rendered correctly * Add dot after url link in models docstrings for consistency
-
- 27 Oct, 2020 1 commit
-
-
F-G Fernandez authored
* style: Added annotation typing for vgg * fix: Fixed annotation typing * refactor: Removed un-necessary import * fix: Added missing annotation for kwargs * fix: Fixed constructor typing * refactor: Refactored typing to minize changes * refactor: Refactored typing cast * fix: Fixed module list typing
-
- 06 Aug, 2019 1 commit
-
-
Francisco Massa authored
* Add docs for video models * Fix docstrings for resnet and vgg
-
- 19 Jul, 2019 1 commit
-
-
apache2046 authored
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)
-
- 04 Jul, 2019 1 commit
-
-
ekka authored
* Add paper references to VGG * Add paper references to ResNet and its variants
-
- 30 Apr, 2019 1 commit
-
-
Philip Meier authored
* added progress flag to model getters * flake8 * bug fix * backward commpability
-
- 14 Feb, 2019 1 commit
-
-
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.
-
- 25 Oct, 2018 1 commit
-
-
Kyryl Truskovskyi authored
* cast mnist targer to int * fix unused variables * fix syntax in tests * remove # noqa; rename l variable to line_split; use except OSError * add W504 * add W504 * add W503,W504 tox.ini
-
- 30 Apr, 2018 1 commit
-
-
Ryuichiro Hataya authored
* 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
-
- 02 Jan, 2018 1 commit
-
-
Yun Chen authored
-
- 26 Aug, 2017 1 commit
-
-
Sri Krishna authored
-
- 31 May, 2017 1 commit
-
-
Sam Gross authored
Fixes #152
-
- 28 Mar, 2017 1 commit
-
-
Karan Dwivedi authored
-
- 16 Mar, 2017 1 commit
-
-
Sam Gross authored
-
- 13 Mar, 2017 1 commit
-
-
Soumith Chintala authored
-
- 27 Feb, 2017 1 commit
-
-
Luke Yeager authored
git ls-files | grep '\.py$' | xargs -n1 -P`nproc` autopep8 -i
-
- 24 Feb, 2017 1 commit
-
-
Luke Yeager authored
-
- 21 Feb, 2017 1 commit
-
-
Sam Gross authored
-
- 02 Feb, 2017 1 commit
-
-
Thibault de Boissiere authored
* Update VGG doco
-
- 17 Jan, 2017 2 commits
-
-
Sam Gross authored
Also add pre-trained ResNet-152 model. ResNet-152: Prec@1 78.312 Prec@5 94.046
-
Soumith Chintala authored
-
- 09 Jan, 2017 1 commit
-
-
Sam Gross authored
-