pytorch
530dcc2b - [ONNX] Tool to verify exported model discrepancy between sets of inputs

Commit
3 years ago
[ONNX] Tool to verify exported model discrepancy between sets of inputs A graph is exported for each set of inputs. The exported graphs are then compared to each other, and discrepancies are reported. This function first checks the jit graph, and then the onnx graph. Unless otherwise specified, the jit/ONNX graph is expected to be the same, regardless of the inputs it used for exporting. A discrepancy would imply the graph exported is not accurate when running with other set of inputs, which will typically results in runtime error or output mismatches. Pull Request resolved: https://github.com/pytorch/pytorch/pull/78323 Approved by: https://github.com/justinchuby, https://github.com/garymm
Author
Committer
Parents
Loading