1. 02 Jan, 2020 1 commit
  2. 19 Dec, 2019 6 commits
  3. 17 Dec, 2019 1 commit
  4. 16 Dec, 2019 4 commits
  5. 12 Dec, 2019 1 commit
  6. 11 Dec, 2019 1 commit
  7. 10 Dec, 2019 2 commits
  8. 06 Dec, 2019 1 commit
  9. 05 Dec, 2019 5 commits
  10. 04 Dec, 2019 3 commits
    • Ankit Jha's avatar
      Add scriptable transform: center_crop, five crop and ten_crop (#1615) · cec7ea72
      Ankit Jha authored
      * add scriptable transform: center_crop
      
      * add test: center_crop
      
      * add scriptable transform: five_crop
      
      * add scriptable transform: five_crop
      
      * add scriptable transform: fix minor issues
      cec7ea72
    • Gerald Baier's avatar
      update dead LSUN link (#1626) · e3a13055
      Gerald Baier authored
      The current link to the dataset is dead. The change links to the dataset's author's personal page, which describe the dataset and is also referenced at https://github.com/fyu/lsun.
      e3a13055
    • pedrofreire's avatar
      Add Deformable Convolution operation. (#1586) · 52b8685b
      pedrofreire authored
      * Add Deformable Convolution operation.
      
      This adds the deformable convolution operation, as described in Deformable Convolutional Networks (https://arxiv.org/abs/1703.06211).
      
      - The code is based on https://github.com/open-mmlab/mmdetection/blob/master/mmdet/ops/dcn/src/deform_conv_cuda.cpp ; the whole code was modified and refactored to remove redundancies and increase clarity, and to adapt it to torchvision.
      
      - The CPU part is a direct copy of the CUDA code; it might make sense to do follow-up adjustments in the CPU code to simplify it / optimize it, or to reuse functionality between CPU and CUDA..
      
      - We also add tests (with a non-trivial set of parameters); they can be made more robust by randomizing the parameters and executing multiple times.
      
      * Update DeformConv to be more consistent w/ Conv2d
      
      * rename some variables and arguments to match Conv2d;
      * add optional bias;
      * add weight, offset and bias as module parameters;
      * remove the n_para...
      52b8685b
  11. 02 Dec, 2019 1 commit
  12. 30 Nov, 2019 1 commit
    • driazati's avatar
      Add tests for results in script vs eager mode (#1430) · 227027d5
      driazati authored
      * Add tests for results in script vs eager mode
      
      This copies some logic from `test_jit.py` to check that a TorchScript'ed
      model's outputs are the same as outputs from the model in eager mode.
      
      To support differences in TorchScript / eager mode outputs, an
      `unwrapper` function can be provided per-model.
      
      * Fix inception, use PYTORCH_TEST_WITH_SLOW
      
      * Update
      
      * Remove assertNestedTensorObjectsEqual
      
      * Add PYTORCH_TEST_WITH_SLOW to CircleCI config
      
      * Add MaskRCNN unwrapper
      
      * fix prec args
      
      * Remove CI changes
      
      * update
      
      * Update
      
      * remove expect changes
      
      * Fix tolerance bug
      
      * Fix breakages
      
      * Fix quantized resnet
      
      * Fix merge errors and simplify code
      
      * DeepLabV3 has been fixed
      
      * Temporarily disable jit compilation
      227027d5
  13. 26 Nov, 2019 3 commits
  14. 25 Nov, 2019 3 commits
    • Yoshitomo Matsubara's avatar
      update default parameters (#1611) · 537f0df7
      Yoshitomo Matsubara authored
      537f0df7
    • eellison's avatar
      Make maskrcnn scriptable (#1407) · d88d8961
      eellison authored
      * almost working...
      
      * respond to comments
      
      * add empty tensor op, handle different output types in generalized rcnn
      
      * clean ups
      
      * address comments
      
      * more changes
      
      * it's working!
      
      * torchscript bugs
      
      * add script/ eager test
      
      * eval script model
      
      * fix flake
      
      * division import
      
      * py2 compat
      
      * update test, fix arange bug
      
      * import division statement
      
      * fix linter
      
      * fixes
      
      * changes needed for JIT master
      
      * cleanups
      
      * remove imagelist_to
      
      * requested changes
      
      * Make FPN backwards-compatible and torchscript compatible
      
      We remove support for feature channels=0, but support for it was already a bit limited
      
      * Fix ONNX regression
      d88d8961
    • Will Brennan's avatar
  15. 21 Nov, 2019 1 commit
  16. 18 Nov, 2019 2 commits
  17. 15 Nov, 2019 4 commits