1. 30 Sep, 2021 4 commits
  2. 29 Sep, 2021 9 commits
    • Kai Zhang's avatar
      Add RegNet Architecture in TorchVision (#4403) · 194a0846
      Kai Zhang authored
      * initial code
      
      * add SqueezeExcitation
      
      * initial code
      
      * add SqueezeExcitation
      
      * add SqueezeExcitation
      
      * regnet blocks, stems and model definition
      
      * nit
      
      * add fc layer
      
      * use Callable instead of Enum for block, stem and activation
      
      * add regnet_x and regnet_y model build functions, add docs
      
      * remove unused depth
      
      * use BN/activation constructor and ConvBNActivation
      
      * add expected test pkl files
      
      * allow custom activation in SqueezeExcitation
      
      * use ReLU as the default activation
      
      * initial code
      
      * add SqueezeExcitation
      
      * initial code
      
      * add SqueezeExcitation
      
      * add SqueezeExcitation
      
      * regnet blocks, stems and model definition
      
      * nit
      
      * add fc layer
      
      * use Callable instead of Enum for block, stem and activation
      
      * add regnet_x and regnet_y model build functions, add docs
      
      * remove unused depth
      
      * use BN/activation constructor and ConvBNActivation
      
      * reuse SqueezeExcitation from efficientnet
      
      * refactor RegNetParams into BlockParams
      
      * use nn.init, replace np with torch
      
      * update README
      
      * construct model with stem, block, classifier instances
      
      * Revert "construct model with stem, block, classifier instances"
      
      This reverts commit 850f5f3ed01a2a9b36fcbf8405afd6e41d2e58ef.
      
      * remove unused blocks
      
      * support scaled model
      
      * fuse into ConvBNActivation
      
      * make reset_parameters private
      
      * fix type errors
      
      * fix for unit test
      
      * add pretrained weights for 6 variant models, update docs
      194a0846
    • Nicolas Hug's avatar
      c4dc3e02
    • Vasilis Vryniotis's avatar
      Replace MobileNetV3's SqueezeExcitation with EfficientNet's one (#4487) · ff126ae2
      Vasilis Vryniotis authored
      * Reuse EfficientNet SE layer.
      
      * Deprecating the mobilenetv3.SqueezeExcitation layer.
      
      * Passing the right activation on quantization.
      
      * Making strict named param.
      
      * Set default params if missing.
      
      * Fixing typos.
      ff126ae2
    • Nicolas Hug's avatar
      Prevent tests from leaking their respective RNG (#4497) · 13bd09dd
      Nicolas Hug authored
      
      
      * Add autouse fixture to save and reset RNG in tests
      
      * Add other RNG generators
      
      * delete freeze_rng_state
      
      * Hopefully fix GaussianBlur test
      
      * Alternative fix, probably better
      
      * revert changes to test_models
      Co-authored-by: default avatarFrancisco Massa <fvsmassa@gmail.com>
      13bd09dd
    • Nicolas Hug's avatar
      5e8a2116
    • Aditya Oke's avatar
      Add JIT tests (#4472) · b5d81f02
      Aditya Oke authored
      
      Co-authored-by: default avatarFrancisco Massa <fvsmassa@gmail.com>
      b5d81f02
    • Aditya Oke's avatar
      Rewrite gallery example for masks to boxes. (#4484) · a9d710a8
      Aditya Oke authored
      
      
      * start writing example
      
      * Update example
      
      * Add PenFudan Files
      
      * Update example
      
      * Remove unused files
      
      * Update file and adopt changes
      
      * Create links, fix float
      Co-authored-by: default avatarNicolas Hug <contact@nicolas-hug.com>
      a9d710a8
    • Philip Meier's avatar
      cleanup prototype datasets (#4471) · a068602e
      Philip Meier authored
      
      
      * cleanup image folder
      
      * make shuffling mandatory
      
      * rename parameter in home() function
      
      * don't show builtin list
      
      * make categories optional in dataset info
      
      * use pseudo-infinite buffer size for shuffler
      Co-authored-by: default avatarFrancisco Massa <fvsmassa@gmail.com>
      a068602e
    • Yi Zhang's avatar
      enable windows cuda unit tests (#4421) · 932ca5a3
      Yi Zhang authored
      
      
      * enable cuda for unittest
      
      * add one check
      
      * rename
      
      * minor fix
      
      * add more check
      
      * add cuda envs
      
      * add more cuda envs
      
      * add driver update
      
      * update vs integration
      
      * get cuda version
      
      * merge
      
      * add envs in install.sh
      
      * reduce some change and add comments
      
      * minor improve
      
      * fix typo
      
      * exit if driver update failed
      
      * fix lint
      
      * fix lint
      
      * Avoid catching exception to show error message
      
      * try 11.1
      
      * try 11.1
      
      * "try 11.1"
      
      * set the downloaded gpu driver dlls only for 10.2
      
      * check torch.cuda.is_available()
      
      * add display.driver
      
      * Revert "Avoid catching exception to show error message"
      
      This reverts commit 1ce58c50de9fefcc3438d29ff398b941f9b6a757.
      
      * Update .circleci/unittest/windows/scripts/set_cuda_envs.sh
      
      * Update .circleci/unittest/windows/scripts/set_cuda_envs.sh
      
      * Update .circleci/unittest/windows/scripts/set_cuda_envs.sh
      Co-authored-by: default avatarNicolas Hug <nicolashug@fb.com>
      Co-authored-by: default avatarVasilis Vryniotis <datumbox@users.noreply.github.com>
      Co-authored-by: default avatarNikita Shulga <nikita.shulga@gmail.com>
      Co-authored-by: default avatarNikita Shulga <nshulga@fb.com>
      932ca5a3
  3. 28 Sep, 2021 2 commits
  4. 27 Sep, 2021 3 commits
    • Nikita Shulga's avatar
      Bump main trunk version to 0.12 (#4490) · 136a16ea
      Nikita Shulga authored
      136a16ea
    • Loi Ly's avatar
      Added gray image support to `adjust_saturation` function (#4480) · f483e71b
      Loi Ly authored
      * update channels parameter to every calling to check_functional_vs_PIL_vs_scripted
      
      * update adjust_saturation
      
      * update docstrings for functional transformations
      
      * parametrize channels
      
      * update docstring of ColorJitter class
      
      * move channels to class's parameter
      
      * remove testing channels for geometric transforms
      
      * revert redundant changes
      
      * revert redundant changes
      
      * update grayscale test cases for randaugment, autoaugment, trivialaugment
      
      * update docstrings of randaugment, autoaugment, trivialaugment
      
      * update docstring of ColorJitter
      
      * fix adjust_hue's docstring
      
      * change test equal tolerance
      
      * refactor grayscale tests
      
      * make get_grayscale_test_image private
      f483e71b
    • peterbell10's avatar
      Update gpu atomics include path (#4478) · 3e27eb21
      peterbell10 authored
      
      
      * Update gpu atomics include path
      
      THC is being removed, and this header now lives in ATen.
      See pytorch/pytorch#65470.
      
      * Empty commit to trigger build
      Co-authored-by: default avatarVasilis Vryniotis <datumbox@users.noreply.github.com>
      3e27eb21
  5. 26 Sep, 2021 1 commit
  6. 24 Sep, 2021 6 commits
  7. 23 Sep, 2021 1 commit
  8. 22 Sep, 2021 1 commit
  9. 21 Sep, 2021 7 commits
  10. 20 Sep, 2021 2 commits
  11. 17 Sep, 2021 3 commits
  12. 16 Sep, 2021 1 commit