Folders and files

Name		Name	Last commit message	Last commit date
examples		examples
nisp		nisp
.gitignore		.gitignore
LICENSE		LICENSE
README.md		README.md
setup.py		setup.py

README
Apache-2.0 license

This is an official git repository of Neural Inverse Source Problems presented in CoRL 2024. We appreciate Jaxpi for offering a great codebase.

Installation

Ensure that you have Python 3.8 or later installed on your system. Our code is GPU-only. We highly recommend using the most recent versions of JAX and JAX-lib, along with compatible CUDA and cuDNN versions. The code has been tested and confirmed to work with the following versions:

JAX 0.4.16
CUDA 12.1
cuDNN 8.9

git clone https://github.com/MMintLab/NISP
cd NISP
pip install -e .

Quickstart

Please follow the readme.md in each example folder. For instance,

cd examples/membrane_real

To train the model, run the following command:

CUDA_VISIBLE_DEVICES=0 python3 main.py --config configs/train.py