Skip to content

nan check in compare models script

Samuel Van Stroud requested to merge svanstro/salt:svanstro/nan-check into main

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

Merge request reports