Skip to content

GitLab

Switch branch/tag
Find file BlameHistoryPermalink
  • Vasilis Vryniotis's avatar
    Implement is_qat in TorchVision (#5299) · 8a16e12f
    Vasilis Vryniotis authored 
    * Add is_qat support using a method getter

* Switch to an internal _fuse_modules

* Fix linter.

* Pass is_qat=False on PTQ

* Fix bug on ra_sampler flag.

* Set is_qat=True for QAT
    8a16e12f
test_models.py 31.6 KB