nan check in compare models script
Description
Check for nans in the pytorch output during export and in the model comparison script.
This might help @jabarr and @avaitkus (but I don't know if it will catch the FPE thing, we need to test that in the dumper)
Review checklist:
-
CI Passing -
Comments addressed -
Source branch is up to date with target
Edited by Samuel Van Stroud