DataWaveProject / CAM

Community Atmosphere Model
Other
1 stars 0 forks source link

Meta: Ability to use (in FTorch) other neural network architectures e.g. CNN, FNOs #20

Open omarjamil opened 2 weeks ago

jatkinson1000 commented 2 weeks ago

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

jatkinson1000 commented 1 week 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