test/test_models.py · b4a075cbc2779318994cf240d070c7335edcbb2a · OpenDAS / vision

Add test to check that classification models are FX-compatible (#3662) · b4a075cb

Francisco Massa authored Jun 21, 2021



* Add test to check that classification models are FX-compatible

* Replace torch.equal with torch.allclose

* remove skipling
Co-authored-by: Nicolas Hug <nicolashug@fb.com>

b4a075cb

test_models.py 24.7 KB

Replace test_models.py