Open omarjamil opened 2 weeks ago
Suggestion is that we fork Hamid's repo above and create a Torchscript versioin of the CNN and FNO, write a Fortran implementation using the same NetCDF data, and then compare the results.
These are broken into https://github.com/DataWaveProject/CAM/issues/23 and https://github.com/DataWaveProject/CAM/issues/24
Input from DW as to what these would be.
@yqsun91 comments: ANN, CNN, and FNO
@jatkinson1000 comments main thing to check is whether we can run these PyTorch features through TorchScript. To check FNO we would need to check this. @HamidPahlavan has an example of a FNO - if this can be shared then we can explore how easy it is to save this as TorchScript: https://github.com/HamidPahlavan/Nonlocality/blob/main/FNO.ipynb