Skip to content

GitLab

Switch branch/tag
  1. 23 Jun, 2020 1 commit
  3. 22 Jun, 2020 2 commits
  5. 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
  7. 17 Jun, 2020 1 commit
  9. 11 Jun, 2020 3 commits
  11. 10 Jun, 2020 1 commit
  13. 09 Jun, 2020 1 commit
  15. 05 Jun, 2020 1 commit
  17. 04 Jun, 2020 3 commits
  19. 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
  21. 01 Jun, 2020 3 commits
  23. 31 May, 2020 1 commit
  25. 30 May, 2020 1 commit
  27. 29 May, 2020 4 commits
  29. 26 May, 2020 3 commits
  31. 21 May, 2020 3 commits
  33. 20 May, 2020 5 commits
    • Philip Meier's avatar
      install project in editabel mode for type check (#2247) · 12b551e7
      Philip Meier authored
      12b551e7
    • Francisco Massa's avatar
      Deprecate Conv2d, ConvTranspose2d and BatchNorm2d (#2244) · 9055250a
      Francisco Massa authored 
      * Deprecate Conv2d, ConvTranspose2d and BatchNorm

* Fix lint
      9055250a
    • Erik's avatar
      add clarifications to Object detection reference documentation (#2241) · cb655900
      Erik authored 
      * Update README.md

added some clarity to get the examples executable.  Waiting to hear back if instructions should mention to setup COCO dataset

* Update README.md

* Update README.md

* Update README.md

* Update README.md

* Update README.md
      cb655900
    • Negin Raoof's avatar
      [ONNX] Fix export of images with no detection (#2215) · 97e21c10
      Negin Raoof authored 
      * Fixing nms on boxes when no detection

* test

* Fix for scale_factor computation

* remove newline

* Fix for mask_rcnn dynanmic axes

* Clean up

* Update transform.py

* Fix for torchscript

* Fix scripting errors

* Fix annotation

* Fix lint

* Fix annotation

* Fix for interpolate scripting

* Fix for scripting

* refactoring

* refactor the code

* Fix annotation

* Fixed annotations

* Added test for resize

* lint

* format

* bump ORT

* ort-nightly version

* Going to ort 1.1.0

* remove version

* install typing-extension

* Export model for images with no detection

* Upgrade ort nightly

* update ORT

* Update test_onnx.py

* updated tests

* Updated tests

* merge

* Update transforms.py

* Update cityscapes.py

* Update celeba.py

* Update caltech.py

* Update pkg_helpers.bash

* Clean up

* Clean up for dynamic split

* Remove extra casts

* flake8

* Fix for mask rcnn no detection export

* clean up

* Enable mask rcnn tests

* Added test

* update ORT

* Update .travis.yml

* fix annotation

* Clean up roi_heads

* clean up

* clean up misc ops
      97e21c10
    • Mike Ruberry's avatar
      Updates integer division to use floor division operator (#2243) · ba63fbdb
      Mike Ruberry authored 
      Another instance of integer division using the division operator. In this case line 266 already shows the correct formulation, so line 185 only needs the update.
      ba63fbdb
  35. 19 May, 2020 3 commits
  37. 18 May, 2020 2 commits
    • eellison's avatar
      remove script classes (#2210) · 222a599e
      eellison authored 
      
Co-authored-by: default avatareellison <eellison@fb.com>
      222a599e
    • Vasiliy Kuznetsov's avatar
      vision classification QAT tutorial: fix for DDP (redo) (#2230) · 7ed3950e
      Vasiliy Kuznetsov authored 
      Summary:

Redo of https://github.com/pytorch/vision/pull/2191

Makes the classification QAT tutorial not crash when used
with DDP. There were two issues:

1. the model was moved to GPU before the observers were added, and they
are created on CPU. In the context of this repo, the fix is to finalize
the model before moving to GPU. We can potentially follow up with a
better error message in the future, in a separate PR.
2. the QAT conversion was running on the DDP'ed model, which had various
problems. The fix is to unwrap the model from DDP before cloning it for
evaluation.

There is still work to do on verifying that BN is working correctly in
QAT + DDP, but saving that for a separate PR.

Test Plan:

```
python -m torch.distributed.launch --use_env references/classification/train_quantization.py --data-path {path_to_imagenet_1k} --output_dir {output_dir}
```

Reviewers:

Subscribers:

Tasks:

Tags:
      7ed3950e