Models are trained with Heat Plume Prediction.
Steps 1 and 3 of LGCNN (Local Global Convolutional Neural Network) are separate, step 2 is a numerical solver that does not require any trained model.
The vanilla UNet can be applied directly end-to-end, just does not give very good results.
For inference follow the guidelines of Heat Plume Prediction and applied all 3 steps/models sequentially to your input data.
Based on raw data from https://doi.org/10.18419/darus-5063.
Please follow the instructions in README.md to prepare the data for training.