Add support for torch>=2, torchvision>=0.15.2, onnx>=1.14

davidt0x commented 1 year ago

Miscellaneous changes to support latest versions of torch, torchvsion, and ONNX. This adds tests for more torchvision models. I also paired down the torchvision tests to speed things up. There are a couple marked as xfail that I still need to work on.

@pgleeson, I cleaned up some stuff in the CI as well.

Re-enable python 3.10 testing on full package
Re-enable windows testing on full package
Install graphviz with a setup-graphviz action, this works on all platforms it seems.
Consolidate windows and linux\mac testing, these got separated at one point, not sure how.

pgleeson commented 1 year ago

Thanks @davidt0x, will try it out.

davidt0x commented 1 year ago

Hey @pgleeson, which version of onnx were you seeing failures? I tried 1.12 (current is 1.14) and it is passing.

pgleeson commented 1 year ago

@davidt0x I think the issue was pytorch 1.11 with the later versions of the onnx libs was causing the "old" form of the node ids to be generated, see this commit: https://github.com/ModECI/MDF/commit/722222459cab9f52c1c87cc0fba41604475d7a09

ModECI / MDF

Add support for torch>=2, torchvision>=0.15.2, onnx>=1.14 #483