helmnet

helmnet copied to clipboard

Export pre-trained model using ONNX

1

Explore exporting the pre-trained model using the [Open Neural Network Exchange (ONNX) format](https://pytorch.org/tutorials/advanced/super_resolution_with_onnxruntime.html). This could then be imported into, e.g., MATLAB so the model can be directly run. One possible...

btreeby

A 3D version of the helmnet should be considered. This would require the following: - Extending the unet to 3D (`Conv3D`, `MaxPool3D`, etc) - Extend the training data to 3D...

btreeby

The network was initially trained assuming constant density and no absorption. Initial tests show that the network may also generalise (even without re-training) to more general cases. This requires updating...

btreeby

solve the size mismatch issue of the generated *.ckpt files based on …

4

…Luc Giraud(Inria) and Paul Mycek(Cerfacs)'s suggestion

Amber-Heung

Need a test that checks that the weights can load after training (see #21 )

astanziola

* **I'm submitting a ...** - [ ] bug report - [x] feature request - [ ] other (e.g. discussion of ideas) * **What is the current behavior?** We don't...

astanziola

Refactor to use complex numbers

2

According to [this repository for Fourier Neural Operators](https://github.com/zongyi-li/fourier_neural_operator), they've made their code 30% faster by switching to native complex representation in PyTorch, which wasn't available when this work was published....

astanziola