test/test_extended_models.py · e6edcef423e14adb226923889d77bf751956a9cd · OpenDAS / vision

Allow custom docs for Weight enums and Weights fields (#5988) · e6edcef4

Nicolas Hug authored May 11, 2022



* POC

* Update torchvision/models/resnet.py
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>

* Apply suggestions from code review
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>

* Fix tests

* ufmt

* Remove useless docstring
Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>

e6edcef4

test_extended_models.py 9.83 KB

Replace test_extended_models.py