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

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

* Replace torch.equal with torch.allclose

* remove skipling

Co-authored-by: Nicolas Hug <b961f2b01ca19b33b8cc7fe0d8d60c0676a9ec0a@fb.com>
pytorch/vision,pytorch/vision,pytorch/vision,pytorch/vision,pytorch/vision,pytorch/vision