Skip to content

GitLab

Switch branch/tag
  1. 30 Jun, 2020 4 commits
    • Edward Z. Yang's avatar
      Port roi_align to actually use dispatcher (#2366) · 44806038
      Edward Z. Yang authored 
      

* Switch torchvision registrations to new operator registration API.

This is still registering everything as catchalls, so we're really just
moving deck chairs around, but payoff is coming soon.
Signed-off-by: default avatarEdward Z. Yang <ezyang@fb.com>

* Port roi_align to actually use dispatcher
Signed-off-by: default avatarEdward Z. Yang <ezyang@fb.com>
      44806038
    • Tongzhou Wang's avatar
      F_t add factor nonnegativity checks for adjust_* (#2356) · 7fd24918
      Tongzhou Wang authored 
      * F_t add factor nonnegativity checks for adjust_*

* Update functional_tensor.py

* Update functional_tensor.py

* Update functional_tensor.py

* Update functional_tensor.py
      7fd24918
    • vfdev's avatar
      Issue 2350 - support of all padding modes with tensors (#2368) · 6fe11d5c
      vfdev authored 
      * [WIP] functional_tensor supports more padding modes

* [WIP] Support all padding modes

* Removed wip symmetric mode

* Improvements according to the review
      6fe11d5c
    • vfdev's avatar
      Unified Tensor/PIL crop (#2342) · a99b6bd7
      vfdev authored 
      * [WIP] Unified Tensor/PIL crop

* Fixed misplaced type annotation

* Fixed tests
- crop with padding
- other tests using mising private functions: _is_pil_image, _get_image_size

* Unified CenterCrop and F.center_crop
- sorted includes in transforms.py
- used py3 annotations

* Unified FiveCrop and F.five_crop

* Improved tests and docs

* Unified TenCrop and F.ten_crop

* Removed useless typing in functional_pil

* Updated code according to the review
- removed useless torch.jit.export
- added missing typing return type
- fixed F.F_pil._is_pil_image -> F._is_pil_image

* Removed useless torch.jit.export

* Improved code according to the review
      a99b6bd7
  3. 29 Jun, 2020 4 commits
  5. 26 Jun, 2020 4 commits
  7. 23 Jun, 2020 3 commits
  9. 22 Jun, 2020 2 commits
  11. 19 Jun, 2020 1 commit
    • Vitaliy Chiley's avatar
      spelling error (#2333) · cd2b7f07
      Vitaliy Chiley authored 
      I was reading through the code and noticed a few spelling errors
orignal -> original
maintaing -> maintaining
      cd2b7f07
  13. 17 Jun, 2020 1 commit
  15. 11 Jun, 2020 3 commits
  17. 10 Jun, 2020 1 commit
  19. 09 Jun, 2020 1 commit
  21. 05 Jun, 2020 1 commit
  23. 04 Jun, 2020 3 commits
  25. 03 Jun, 2020 1 commit
    • Vasiliy Kuznetsov's avatar
      torchvision QAT tutorial: update for QAT with DDP (#2280) · 39021408
      Vasiliy Kuznetsov authored 
      Summary:

We've made two recent changes to QAT in PyTorch core:
1. add support for SyncBatchNorm
2. make eager mode QAT prepare scripts respect device affinity

This PR updates the torchvision QAT reference script to take
advantage of both of these.  This should be landed after
https://github.com/pytorch/pytorch/pull/39337 (the last PT
fix) to avoid compatibility issues.

Test Plan:

```
python -m torch.distributed.launch
  --nproc_per_node 8
  --use_env
  references/classification/train_quantization.py
  --data-path {imagenet1k_subset}
  --output-dir {tmp}
  --sync-bn
```

Reviewers:

Subscribers:

Tasks:

Tags:
      39021408
  27. 01 Jun, 2020 3 commits
  29. 31 May, 2020 1 commit
  31. 30 May, 2020 1 commit
  33. 29 May, 2020 4 commits
  35. 26 May, 2020 2 commits