- 23 Oct, 2020 1 commit
-
-
F-G Fernandez authored
* fix: Fixed constructor typing in models._utils * fix: Fixed constructor typing in models.alexnet * fix: Fixed constructor typing in models.mnasnet * fix: Fixed constructor typing in models.squeezenet
-
- 22 Oct, 2020 1 commit
-
-
F-G Fernandez authored
* style: Added annotation typing for squeezenet * feat: Added typing for kwargs
-
- 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)
-
- 03 May, 2019 1 commit
-
-
Vitor Finotti Ferreira authored
-
- 30 Apr, 2019 1 commit
-
-
Philip Meier authored
* added progress flag to model getters * flake8 * bug fix * backward commpability
-
- 08 Apr, 2019 1 commit
-
-
Allan Wang authored
-
- 06 Nov, 2018 1 commit
-
-
Adán Mora-Fallas authored
* AdaptiveAvgpool2d integrated The Average Pool Layer on SqueezeNet and ResNet was hardcoded, this was changed, now any the input size is accepted. * error fixed * linting fixed
-
- 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
-
- 22 Oct, 2017 1 commit
-
-
Alykhan Tejani authored
-
- 21 May, 2017 1 commit
-
-
Sri Krishna authored
changed a small mistake.
-
- 30 Apr, 2017 1 commit
-
-
Marat Dukhan 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
-
- 11 Feb, 2017 1 commit
-
-
Marat Dukhan authored
* Add SqueezeNet 1.0 and 1.1 models * Selectively avoid inplace in SqueezeNet * Use Glorot uniform initialization in SqueezeNet * Make all ReLU in SqueezeNet in-place * Add pretrained SqueezeNet 1.0 and 1.1 * Minor fixes in SqueezeNet models
-