Export pre-trained model using ONNX

ucl-bug / helmnet

Deep-learning iterative solver for the heterogeneous 2D Helmholtz equation

MIT License

27 stars 8 forks source link

Export pre-trained model using ONNX #31

Open btreeby opened 1 year ago

btreeby commented 1 year ago

Explore exporting the pre-trained model using the Open Neural Network Exchange (ONNX) format. This could then be imported into, e.g., MATLAB so the model can be directly run. One possible complication is the use of the FFT within the residual calculation, which may not be supported.

astanziola commented 1 year ago

Should be OK to export modules with FFTs: https://github.com/onnx/onnx/pull/3741