DeepGPR 0.0.8__tar.gz → 0.0.10__tar.gz

deepgpr-0.0.8/README.md → deepgpr-0.0.10/PKG-INFO

@@ -1,3 +1,20 @@
+Metadata-Version: 2.4
+Name: DeepGPR
+Version: 0.0.10
+Summary: PyTorch and CUDA for GPR FWI
+Author-email: Lei Liu <liulei990222@gmail.com>
+Classifier: Programming Language :: Python :: 3
+Classifier: Operating System :: Microsoft :: Windows
+Classifier: Operating System :: POSIX :: Linux
+Classifier: Topic :: Scientific/Engineering
+Classifier: Topic :: Scientific/Engineering :: Physics
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+Requires-Dist: numpy
+Requires-Dist: scipy
+Requires-Dist: matplotlib
+Requires-Dist: torch
+
 # DeepGPR
 
 DeepGPR provides a wave propagation module for PyTorch, designed for applications such as Ground Penetrating Radar (GPR) imaging and inversion. Its core concepts are derived from Deepwave. You can use it to perform both forward modeling and backpropagation—thereby enabling the simulation of wave propagation to generate synthetic data—as well as for Full Waveform Inversion (FWI). Furthermore, you can integrate this wave propagation functionality into a larger operational pipeline—incorporating various wavelets, loss functions, and other components—to achieve end-to-end forward and reverse propagation, powered by automatic differentiation and our high-performance operators.
@@ -11,52 +28,26 @@ Gradients of the output receiver data can be computed with respect to model para
 
 Utilizes CPML, allowing the width of the PML layer to be configured independently for each boundary.
 
-All operations are executed on the GPU.
-
-Supports techniques such as checkpointing, DDP, and the utilization of CPU memory to minimize GPU memory consumption, thereby enabling the execution of large-scale models.
-
-
-## System Requirements
-
-- **OS**: Linux and Windows
-- **Python**: Python 3.8+
-- **GPU**: NVIDIA GPU with sufficient VRAM for 2D/3D computational grids
-- **CUDA**: NVIDIA driver and CUDA Toolkit
-- **Libraries**:
-  - `torch` with CUDA support
-  - `numpy`
-  - `scipy`
-  - `matplotlib`
-
-### Additional Requirements for Building from Source
+The compute backend can run on CUDA GPUs or on CPU. The CPU backend is implemented in C and is selected automatically when `device='cpu'`.
 
-DeepGPR contains a CUDA backend that is compiled during installation. Therefore, users who install the package from source need a working CUDA/C++ build environment.
+The FDTD spatial finite-difference order can be selected with `fdtd_order=2`, `4`, or `8` (default: `2`).
 
-#### Linux
+The FWI gradient mode can be selected with `mode=2` or `mode=3`. `mode=2` keeps the previous Ez-only gradient behavior, while `mode=3` uses Ex, Ey, and Ez forward/adjoint electric-field contributions for relative permittivity and conductivity gradients.
 
-- CUDA Toolkit with `nvcc`
-- GCC/G++ compiler compatible with the installed CUDA version
-- CMake 3.23 or later
+Supports techniques such as checkpointing, DDP, and the utilization of CPU memory to minimize GPU memory consumption, thereby enabling the execution of large-scale models.
 
-#### Windows
 
-- CUDA Toolkit with `nvcc`
-- CMake 3.23 or later
-- Microsoft Visual Studio Build Tools or Visual Studio Community
-- The **Desktop development with C++** workload
-- MSVC C++ compiler `cl.exe`
-- Windows SDK
-- Optional: `ninja`
+## System Requirements
 
-On Windows, `nvcc` relies on the MSVC compiler to build CUDA code. Please make sure that both `nvcc` and `cl.exe` are available before running:
+- **OS**: Linux, Windows, and macOS for CPU execution; Linux and Windows for CUDA execution
+- **Environment**: Python 3.8+, CUDA Toolkit for CUDA execution
+- **Libraries**: `torch`, `numpy`, `scipy`, `matplotlib`
+- **Hardware**: NVIDIA GPU with sufficient VRAM for CUDA execution; CPU execution works without a GPU.
 
-```bash
-pip install .
-```
 
 ## Start
 
-Before use, you must ensure that you have an NVIDIA graphics card and have installed a CUDA-enabled version of PyTorch.
+Before CUDA use, you must ensure that you have an NVIDIA graphics card and have installed a CUDA-enabled version of PyTorch. For CPU use, install a CPU build of PyTorch and include a compiled `deepgpr_cpu` shared library in `src/DeepGPR/lib`.
 
 DeepGPR can then be installed using
 
@@ -74,7 +65,7 @@ import DeepGPR
 import matplotlib.pyplot as plt
 
 # Set up the parameters and models
-device=torch.device("cuda")
+device=torch.device("cuda" if torch.cuda.is_available() else "cpu")
 dx=0.02
 dt=3e-11
 nt=2000
@@ -96,7 +87,8 @@ r = DeepGPR.compute(
     source_amplitudes=source_amplitudes,
     source_location=source_location, 
     receiver_location=receiver_location, 
-    er=er, se=se
+    er=er, se=se,
+    fdtd_order=2
 )
 
 (r[-1]**2).sum().backward()
@@ -150,16 +142,20 @@ def compute(device, dx=None, dt=None,
             E=None, H=None, PML=None,
             pmlthick=10, source_direction=2, reciever_direction=2,
             model_gradient_sampling_interval=1,
-            use_async_offload=False):
+            use_async_offload=False,
+            fdtd_order=2,
+            mode=2):
 ```
 ## 📥 Input Parameters
 ### 1. Basic Physics & Grid Parameters
 
 | Parameter | Data Type | Description |
 | :--- | :--- | :--- |
-| **`device`** | `torch.device` / `str` | PyTorch computation device, e.g., `'cuda:0'`. Determines where the computation takes place. |
+| **`device`** | `torch.device` / `str` | PyTorch computation device, e.g., `'cuda:0'` or `'cpu'`. CUDA loads `deepgpr.so/.dll`; CPU loads `deepgpr_cpu.so/.dll/.dylib`. |
 | **`dx`** | `float` | Spatial grid step size (assuming an isotropic grid, i.e., $dx = dy = dz$). Typically in meters (m). |
 | **`dt`** | `float` | Time step size. **Note**: Must strictly satisfy the CFL (Courant-Friedrichs-Lewy) stability condition, or an exception will be raised. Typically in seconds (s). |
+| **`fdtd_order`** | `int` | Spatial finite-difference order used by the FDTD field updates. Supported values are `2`, `4`, and `8`; default is `2` for compatibility with earlier versions. |
+| **`mode`** | `int` | FWI gradient mode. `2` keeps the previous Ez-only model-gradient calculation. `3` uses Ex, Ey, and Ez electric-field contributions for relative permittivity and conductivity gradients. |
 ### 2. Medium Model Parameters
 
 This section defines the electromagnetic properties of the simulation space. For 2D simulations, set `nz=1`.
@@ -178,7 +174,7 @@ This section defines the geometric observation system (coordinates) and the exci
 
 | Parameter | Data Type | Shape | Description |
 | :--- | :--- | :--- | :--- |
-| **`source_amplitudes`** | `Tensor` (float) | `(num_waveforms, nt)` | Source excitation waveforms. `nt` is the total number of time steps.<br>- If `num_waveforms == 1`: All sources share this single waveform.<br>- If `num_waveforms == nsr`: Each source uses its corresponding waveform. |
+| **`source_amplitudes`** | `Tensor` (float) | `(num_waveforms, nt, 1)` | Source excitation waveforms. `nt` is the total number of time steps.<br>- If `num_waveforms == 1`: All sources share this single waveform.<br>- If `num_waveforms == nsr`: Each source uses its corresponding waveform. |
 | **`source_location`** | `Tensor` (int) | `(nstep, nsr, 3)` | Grid coordinate indices of the sources.<br>The last dimension corresponds to `[x_idx, y_idx, z_idx]`. |
 | **`receiver_location`** | `Tensor` (int) | `(nstep, nrx, 3)` | Grid coordinate indices of the receivers.<br>The last dimension corresponds to `[x_idx, y_idx, z_idx]`. |
 | **`source_direction`** | `int` | Scalar | Polarization direction/component of the source excitation.<br>`0` = X, `1` = Y, `2` = Z (e.g., exciting $E_z$). |
@@ -196,7 +192,34 @@ This section defines the geometric observation system (coordinates) and the exci
 | :--- | :--- | :--- | :--- |
 | **`pmlthick`** | `int` / `list` / `Tensor`| Scalar or list of 6 | Thickness (in grid layers) of the PML (Perfectly Matched Layer) absorbing boundaries.<br>- Integer `p`: All six boundaries have thickness `p` (Z-boundaries are ignored in 2D).<br>- List `[x0, xm, y0, ym, z0, zm]`: Specific thicknesses for the 6 boundaries. |
 | **`model_gradient_sampling_interval`**| `int` | Scalar | Wavefield sampling interval during forward propagation (Default: 1).<br>A larger integer reduces the VRAM usage for the saved `Eall` tensor, but may decrease the accuracy of backpropagated gradients. |
-| **`use_async_offload`** | `bool` | Scalar | VRAM optimization flag (Default: `False`).<br>If `True`, the internal full wavefield tensor (`Eall`) is asynchronously offloaded to page-locked host memory (`pin_memory` CPU RAM). This drastically reduces GPU VRAM consumption at the cost of slightly slower computation times due to PCIe data transfer latency. |
+| **`use_async_offload`** | `bool` | Scalar | CUDA-only VRAM optimization flag (Default: `False`).<br>If `True`, the internal full wavefield tensor (`Eall`) is asynchronously offloaded to page-locked host memory (`pin_memory` CPU RAM). This drastically reduces GPU VRAM consumption at the cost of slightly slower computation times due to PCIe data transfer latency. On CPU this option is ignored. |
+
+### 4.1 FWI Gradient Mode
+
+`mode` only changes how the model gradients are accumulated during backpropagation:
+
+- `mode=2` (default): Saves Ez in `Eall` and computes relative permittivity/conductivity gradients from Ez only. This keeps the old behavior.
+- `mode=3`: Saves Ex, Ey, and Ez in `Eall` and computes relative permittivity/conductivity gradients from all three electric-field components. This is intended for complete 3D Maxwell FWI. The adjoint source polarization is not changed by this option.
+
+## CPU Backend Build
+
+The CPU backend is a plain C shared library and is built with OpenMP by default. Build it into `src/DeepGPR/lib` before running with `device='cpu'`. You can control CPU thread count with `OMP_NUM_THREADS`.
+
+```bash
+# Linux
+cc -std=c99 -O3 -fopenmp -fPIC -shared -o src/DeepGPR/lib/deepgpr_cpu.so src/DeepGPR/lib/deepgpr_cpu.c
+
+# macOS
+brew install libomp
+LIBOMP_PREFIX="$(brew --prefix libomp)"
+cc -std=c99 -O3 -Xpreprocessor -fopenmp -DDEEPGPR_USE_OPENMP -I"$LIBOMP_PREFIX/include" -Wl,-rpath,"$LIBOMP_PREFIX/lib" -L"$LIBOMP_PREFIX/lib" -fPIC -shared -o src/DeepGPR/lib/deepgpr_cpu.dylib src/DeepGPR/lib/deepgpr_cpu.c -lomp
+```
+
+On Windows, build `src\DeepGPR\lib\deepgpr_cpu.dll` with MSVC:
+
+```powershell
+cl /LD /O2 /openmp /Fe:src\DeepGPR\lib\deepgpr_cpu.dll src\DeepGPR\lib\deepgpr_cpu.c
+```
 
 ### 5. Field Variable States (Checkpoints / Initial Fields)
 
@@ -218,8 +241,10 @@ The function returns a tuple of 5 elements. These are used to extract synthetic
 return Eall, (Ex, Ey, Ez), (Hx, Hy, Hz), (x0EPhi1...zmHPhi2), receiver_amplitudes
 ```
 
-1.  **`Eall`**: The global electric field history saved for gradient calculation.
-    *   **Shape**: `(nt_saved, nstep, nx, ny, nz)` (where `nt_saved` depends on `nt` and `model_gradient_sampling_interval`).
+1.  **`Eall`**: The electric field history saved for gradient calculation.
+    *   **Shape when `mode=2`**: `(nt_saved, nstep, nx, ny, nz)`, storing Ez only.
+    *   **Shape when `mode=3`**: `(3, nt_saved, nstep, nx, ny, nz)`, storing components in `[Ex, Ey, Ez]` order.
+    *   `nt_saved` depends on `nt` and `model_gradient_sampling_interval`.
 2.  **`(Ex, Ey, Ez)`**: The 3D electric field state at the final time step. 
 3.  **`(Hx, Hy, Hz)`**: The 3D magnetic field state at the final time step.
 4.  **`(PML_Tuple)`**: A tuple of 24 Tensors recording the final time step state of the PML auxiliary $\Phi$ variables.
@@ -244,4 +269,4 @@ If you find our codes useful, please kindly cite this article. Thanks.
 
   publisher={Elsevier}
 
-}
+}

deepgpr-0.0.8/PKG-INFO → deepgpr-0.0.10/README.md

@@ -1,17 +1,3 @@
-Metadata-Version: 2.2
-Name: DeepGPR
-Version: 0.0.8
-Summary: PyTorch and CUDA for GPR FWI
-Author-Email: Lei Liu <liulei990222@gmail.com>
-Classifier: Programming Language :: Python :: 3
-Classifier: Operating System :: Microsoft :: Windows
-Classifier: Operating System :: POSIX :: Linux
-Requires-Python: >=3.8
-Requires-Dist: numpy
-Requires-Dist: scipy
-Requires-Dist: matplotlib
-Description-Content-Type: text/markdown
-
 # DeepGPR
 
 DeepGPR provides a wave propagation module for PyTorch, designed for applications such as Ground Penetrating Radar (GPR) imaging and inversion. Its core concepts are derived from Deepwave. You can use it to perform both forward modeling and backpropagation—thereby enabling the simulation of wave propagation to generate synthetic data—as well as for Full Waveform Inversion (FWI). Furthermore, you can integrate this wave propagation functionality into a larger operational pipeline—incorporating various wavelets, loss functions, and other components—to achieve end-to-end forward and reverse propagation, powered by automatic differentiation and our high-performance operators.
@@ -25,52 +11,26 @@ Gradients of the output receiver data can be computed with respect to model para
 
 Utilizes CPML, allowing the width of the PML layer to be configured independently for each boundary.
 
-All operations are executed on the GPU.
-
-Supports techniques such as checkpointing, DDP, and the utilization of CPU memory to minimize GPU memory consumption, thereby enabling the execution of large-scale models.
-
-
-## System Requirements
-
-- **OS**: Linux and Windows
-- **Python**: Python 3.8+
-- **GPU**: NVIDIA GPU with sufficient VRAM for 2D/3D computational grids
-- **CUDA**: NVIDIA driver and CUDA Toolkit
-- **Libraries**:
-  - `torch` with CUDA support
-  - `numpy`
-  - `scipy`
-  - `matplotlib`
-
-### Additional Requirements for Building from Source
+The compute backend can run on CUDA GPUs or on CPU. The CPU backend is implemented in C and is selected automatically when `device='cpu'`.
 
-DeepGPR contains a CUDA backend that is compiled during installation. Therefore, users who install the package from source need a working CUDA/C++ build environment.
+The FDTD spatial finite-difference order can be selected with `fdtd_order=2`, `4`, or `8` (default: `2`).
 
-#### Linux
+The FWI gradient mode can be selected with `mode=2` or `mode=3`. `mode=2` keeps the previous Ez-only gradient behavior, while `mode=3` uses Ex, Ey, and Ez forward/adjoint electric-field contributions for relative permittivity and conductivity gradients.
 
-- CUDA Toolkit with `nvcc`
-- GCC/G++ compiler compatible with the installed CUDA version
-- CMake 3.23 or later
+Supports techniques such as checkpointing, DDP, and the utilization of CPU memory to minimize GPU memory consumption, thereby enabling the execution of large-scale models.
 
-#### Windows
 
-- CUDA Toolkit with `nvcc`
-- CMake 3.23 or later
-- Microsoft Visual Studio Build Tools or Visual Studio Community
-- The **Desktop development with C++** workload
-- MSVC C++ compiler `cl.exe`
-- Windows SDK
-- Optional: `ninja`
+## System Requirements
 
-On Windows, `nvcc` relies on the MSVC compiler to build CUDA code. Please make sure that both `nvcc` and `cl.exe` are available before running:
+- **OS**: Linux, Windows, and macOS for CPU execution; Linux and Windows for CUDA execution
+- **Environment**: Python 3.8+, CUDA Toolkit for CUDA execution
+- **Libraries**: `torch`, `numpy`, `scipy`, `matplotlib`
+- **Hardware**: NVIDIA GPU with sufficient VRAM for CUDA execution; CPU execution works without a GPU.
 
-```bash
-pip install .
-```
 
 ## Start
 
-Before use, you must ensure that you have an NVIDIA graphics card and have installed a CUDA-enabled version of PyTorch.
+Before CUDA use, you must ensure that you have an NVIDIA graphics card and have installed a CUDA-enabled version of PyTorch. For CPU use, install a CPU build of PyTorch and include a compiled `deepgpr_cpu` shared library in `src/DeepGPR/lib`.
 
 DeepGPR can then be installed using
 
@@ -88,7 +48,7 @@ import DeepGPR
 import matplotlib.pyplot as plt
 
 # Set up the parameters and models
-device=torch.device("cuda")
+device=torch.device("cuda" if torch.cuda.is_available() else "cpu")
 dx=0.02
 dt=3e-11
 nt=2000
@@ -110,7 +70,8 @@ r = DeepGPR.compute(
     source_amplitudes=source_amplitudes,
     source_location=source_location, 
     receiver_location=receiver_location, 
-    er=er, se=se
+    er=er, se=se,
+    fdtd_order=2
 )
 
 (r[-1]**2).sum().backward()
@@ -164,16 +125,20 @@ def compute(device, dx=None, dt=None,
             E=None, H=None, PML=None,
             pmlthick=10, source_direction=2, reciever_direction=2,
             model_gradient_sampling_interval=1,
-            use_async_offload=False):
+            use_async_offload=False,
+            fdtd_order=2,
+            mode=2):
 ```
 ## 📥 Input Parameters
 ### 1. Basic Physics & Grid Parameters
 
 | Parameter | Data Type | Description |
 | :--- | :--- | :--- |
-| **`device`** | `torch.device` / `str` | PyTorch computation device, e.g., `'cuda:0'`. Determines where the computation takes place. |
+| **`device`** | `torch.device` / `str` | PyTorch computation device, e.g., `'cuda:0'` or `'cpu'`. CUDA loads `deepgpr.so/.dll`; CPU loads `deepgpr_cpu.so/.dll/.dylib`. |
 | **`dx`** | `float` | Spatial grid step size (assuming an isotropic grid, i.e., $dx = dy = dz$). Typically in meters (m). |
 | **`dt`** | `float` | Time step size. **Note**: Must strictly satisfy the CFL (Courant-Friedrichs-Lewy) stability condition, or an exception will be raised. Typically in seconds (s). |
+| **`fdtd_order`** | `int` | Spatial finite-difference order used by the FDTD field updates. Supported values are `2`, `4`, and `8`; default is `2` for compatibility with earlier versions. |
+| **`mode`** | `int` | FWI gradient mode. `2` keeps the previous Ez-only model-gradient calculation. `3` uses Ex, Ey, and Ez electric-field contributions for relative permittivity and conductivity gradients. |
 ### 2. Medium Model Parameters
 
 This section defines the electromagnetic properties of the simulation space. For 2D simulations, set `nz=1`.
@@ -192,7 +157,7 @@ This section defines the geometric observation system (coordinates) and the exci
 
 | Parameter | Data Type | Shape | Description |
 | :--- | :--- | :--- | :--- |
-| **`source_amplitudes`** | `Tensor` (float) | `(num_waveforms, nt)` | Source excitation waveforms. `nt` is the total number of time steps.<br>- If `num_waveforms == 1`: All sources share this single waveform.<br>- If `num_waveforms == nsr`: Each source uses its corresponding waveform. |
+| **`source_amplitudes`** | `Tensor` (float) | `(num_waveforms, nt, 1)` | Source excitation waveforms. `nt` is the total number of time steps.<br>- If `num_waveforms == 1`: All sources share this single waveform.<br>- If `num_waveforms == nsr`: Each source uses its corresponding waveform. |
 | **`source_location`** | `Tensor` (int) | `(nstep, nsr, 3)` | Grid coordinate indices of the sources.<br>The last dimension corresponds to `[x_idx, y_idx, z_idx]`. |
 | **`receiver_location`** | `Tensor` (int) | `(nstep, nrx, 3)` | Grid coordinate indices of the receivers.<br>The last dimension corresponds to `[x_idx, y_idx, z_idx]`. |
 | **`source_direction`** | `int` | Scalar | Polarization direction/component of the source excitation.<br>`0` = X, `1` = Y, `2` = Z (e.g., exciting $E_z$). |
@@ -210,7 +175,34 @@ This section defines the geometric observation system (coordinates) and the exci
 | :--- | :--- | :--- | :--- |
 | **`pmlthick`** | `int` / `list` / `Tensor`| Scalar or list of 6 | Thickness (in grid layers) of the PML (Perfectly Matched Layer) absorbing boundaries.<br>- Integer `p`: All six boundaries have thickness `p` (Z-boundaries are ignored in 2D).<br>- List `[x0, xm, y0, ym, z0, zm]`: Specific thicknesses for the 6 boundaries. |
 | **`model_gradient_sampling_interval`**| `int` | Scalar | Wavefield sampling interval during forward propagation (Default: 1).<br>A larger integer reduces the VRAM usage for the saved `Eall` tensor, but may decrease the accuracy of backpropagated gradients. |
-| **`use_async_offload`** | `bool` | Scalar | VRAM optimization flag (Default: `False`).<br>If `True`, the internal full wavefield tensor (`Eall`) is asynchronously offloaded to page-locked host memory (`pin_memory` CPU RAM). This drastically reduces GPU VRAM consumption at the cost of slightly slower computation times due to PCIe data transfer latency. |
+| **`use_async_offload`** | `bool` | Scalar | CUDA-only VRAM optimization flag (Default: `False`).<br>If `True`, the internal full wavefield tensor (`Eall`) is asynchronously offloaded to page-locked host memory (`pin_memory` CPU RAM). This drastically reduces GPU VRAM consumption at the cost of slightly slower computation times due to PCIe data transfer latency. On CPU this option is ignored. |
+
+### 4.1 FWI Gradient Mode
+
+`mode` only changes how the model gradients are accumulated during backpropagation:
+
+- `mode=2` (default): Saves Ez in `Eall` and computes relative permittivity/conductivity gradients from Ez only. This keeps the old behavior.
+- `mode=3`: Saves Ex, Ey, and Ez in `Eall` and computes relative permittivity/conductivity gradients from all three electric-field components. This is intended for complete 3D Maxwell FWI. The adjoint source polarization is not changed by this option.
+
+## CPU Backend Build
+
+The CPU backend is a plain C shared library and is built with OpenMP by default. Build it into `src/DeepGPR/lib` before running with `device='cpu'`. You can control CPU thread count with `OMP_NUM_THREADS`.
+
+```bash
+# Linux
+cc -std=c99 -O3 -fopenmp -fPIC -shared -o src/DeepGPR/lib/deepgpr_cpu.so src/DeepGPR/lib/deepgpr_cpu.c
+
+# macOS
+brew install libomp
+LIBOMP_PREFIX="$(brew --prefix libomp)"
+cc -std=c99 -O3 -Xpreprocessor -fopenmp -DDEEPGPR_USE_OPENMP -I"$LIBOMP_PREFIX/include" -Wl,-rpath,"$LIBOMP_PREFIX/lib" -L"$LIBOMP_PREFIX/lib" -fPIC -shared -o src/DeepGPR/lib/deepgpr_cpu.dylib src/DeepGPR/lib/deepgpr_cpu.c -lomp
+```
+
+On Windows, build `src\DeepGPR\lib\deepgpr_cpu.dll` with MSVC:
+
+```powershell
+cl /LD /O2 /openmp /Fe:src\DeepGPR\lib\deepgpr_cpu.dll src\DeepGPR\lib\deepgpr_cpu.c
+```
 
 ### 5. Field Variable States (Checkpoints / Initial Fields)
 
@@ -232,8 +224,10 @@ The function returns a tuple of 5 elements. These are used to extract synthetic
 return Eall, (Ex, Ey, Ez), (Hx, Hy, Hz), (x0EPhi1...zmHPhi2), receiver_amplitudes
 ```
 
-1.  **`Eall`**: The global electric field history saved for gradient calculation.
-    *   **Shape**: `(nt_saved, nstep, nx, ny, nz)` (where `nt_saved` depends on `nt` and `model_gradient_sampling_interval`).
+1.  **`Eall`**: The electric field history saved for gradient calculation.
+    *   **Shape when `mode=2`**: `(nt_saved, nstep, nx, ny, nz)`, storing Ez only.
+    *   **Shape when `mode=3`**: `(3, nt_saved, nstep, nx, ny, nz)`, storing components in `[Ex, Ey, Ez]` order.
+    *   `nt_saved` depends on `nt` and `model_gradient_sampling_interval`.
 2.  **`(Ex, Ey, Ez)`**: The 3D electric field state at the final time step. 
 3.  **`(Hx, Hy, Hz)`**: The 3D magnetic field state at the final time step.
 4.  **`(PML_Tuple)`**: A tuple of 24 Tensors recording the final time step state of the PML auxiliary $\Phi$ variables.
@@ -258,4 +252,4 @@ If you find our codes useful, please kindly cite this article. Thanks.
 
   publisher={Elsevier}
 
-}
+}

deepgpr-0.0.10/pyproject.toml

@@ -0,0 +1,48 @@
+[build-system]
+requires = ["setuptools>=70", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "DeepGPR"
+version = "0.0.10"
+authors = [
+    { name = "Lei Liu", email = "liulei990222@gmail.com" }
+]
+description = "PyTorch and CUDA for GPR FWI"
+readme = "README.md"
+requires-python = ">=3.8"
+
+classifiers = [
+    "Programming Language :: Python :: 3",
+    "Operating System :: Microsoft :: Windows",
+    "Operating System :: POSIX :: Linux",
+    "Topic :: Scientific/Engineering",
+    "Topic :: Scientific/Engineering :: Physics"
+]
+
+dependencies = [
+    "numpy",
+    "scipy",
+    "matplotlib",
+    "torch"
+]
+
+[tool.setuptools]
+package-dir = { "" = "src" }
+include-package-data = true
+
+[tool.setuptools.packages.find]
+where = ["src"]
+
+[tool.setuptools.package-data]
+DeepGPR = [
+    "lib/*.cu",
+    "lib/*.cuh",
+    "lib/*.h",
+    "lib/*.hpp",
+    "lib/*.cpp",
+    "lib/*.c",
+    "lib/*.dll",
+    "lib/*.so",
+    "lib/*.pyd"
+]

deepgpr-0.0.10/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+