A deep learning framework for solving the nonlinear hyperbolic systems.
This repository contains the official implementation for the paper: Capturing Shock Waves by Relaxation Neural Networks
This code has been tested and confirmed to work with the following versions:
- PYTHON 3.10.12
- PYTORCH 2.3.0
- NUMPY 1.24.4
Install RelaxNN with the following commands:
git clone [email protected]:mazhengcn/relaxation-nn.git
First, down to the relaxnn
folder:
cd relaxnn
Then, to train our model, run the following command to pass the hyperparams and execute the main.py
:
bash run_main.sh
Finally, to evaluate the model's performance, you can use evaluate.py
.
Our reference data are obtained by Clawpack: https://www.clawpack.org/, the information about the reference data are described in data information.txt
in data
folder.
@misc{zhou2024capturing,
title={Capturing Shock Waves by Relaxation Neural Networks},
author={Nan Zhou and Zheng Ma},
year={2024},
eprint={2404.01163},
archivePrefix={arXiv},
primaryClass={math.NA}
}