Export: print only first line and print error status

justinchuby / torch-onnx

Prototype of the next torch exporter

MIT License

1 stars 1 forks source link

Export: print only first line and print error status #69

Closed justinchuby closed 1 week ago

justinchuby commented 1 week ago

Produce in stdout

Obtain model graph for `MyModel()` with `torch.export.export`...
Obtain model graph for `MyModel()` with `torch.export.export`... ✅
Translate the graph into ONNX...
Translate the graph into ONNX... ✅
Run `onnx.checker` on the ONNX model...
Run `onnx.checker` on the ONNX model... ❌