Fourier Neural Operater for Darcy Flow
This example demonstrates how to set up a data-driven model for a 2D
Darcy flow using the Fourier Neural Operator (FNO) architecture inside
of PhysicsNeMo. This example runs on a single GPU, go to the
darcy_nested_fno example for exploring a multi-GPU training.
Prerequisites
Install the required dependencies by running below:
pip install -r requirements.txt
To train the model, run
python train_fno_darcy.py
training data will be generated on the fly.