wavedl 1.3.0__py3-none-any.whl → 1.4.0__py3-none-any.whl

{wavedl-1.3.0.dist-info → wavedl-1.4.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: wavedl
-Version: 1.3.0
+Version: 1.4.0
 Summary: A Scalable Deep Learning Framework for Wave-Based Inverse Problems
 Author: Ductho Le
 License: MIT
@@ -30,31 +30,18 @@ Requires-Dist: scikit-learn>=1.2.0
 Requires-Dist: pandas>=2.0.0
 Requires-Dist: matplotlib>=3.7.0
 Requires-Dist: tqdm>=4.65.0
-Requires-Dist: wandb>=0.15.0
 Requires-Dist: pyyaml>=6.0.0
 Requires-Dist: h5py>=3.8.0
 Requires-Dist: safetensors>=0.3.0
-Provides-Extra: dev
-Requires-Dist: pytest>=7.0.0; extra == "dev"
-Requires-Dist: pytest-xdist>=3.5.0; extra == "dev"
-Requires-Dist: ruff>=0.8.0; extra == "dev"
-Requires-Dist: pre-commit>=3.5.0; extra == "dev"
-Provides-Extra: onnx
-Requires-Dist: onnx>=1.14.0; extra == "onnx"
-Requires-Dist: onnxruntime>=1.15.0; extra == "onnx"
-Provides-Extra: compile
-Requires-Dist: triton; sys_platform == "linux" and extra == "compile"
-Provides-Extra: hpo
-Requires-Dist: optuna>=3.0.0; extra == "hpo"
-Provides-Extra: all
-Requires-Dist: pytest>=7.0.0; extra == "all"
-Requires-Dist: pytest-xdist>=3.5.0; extra == "all"
-Requires-Dist: ruff>=0.8.0; extra == "all"
-Requires-Dist: pre-commit>=3.5.0; extra == "all"
-Requires-Dist: onnx>=1.14.0; extra == "all"
-Requires-Dist: onnxruntime>=1.15.0; extra == "all"
-Requires-Dist: triton; sys_platform == "linux" and extra == "all"
-Requires-Dist: optuna>=3.0.0; extra == "all"
+Requires-Dist: wandb>=0.15.0
+Requires-Dist: optuna>=3.0.0
+Requires-Dist: onnx>=1.14.0
+Requires-Dist: onnxruntime>=1.15.0
+Requires-Dist: pytest>=7.0.0
+Requires-Dist: pytest-xdist>=3.5.0
+Requires-Dist: ruff>=0.8.0
+Requires-Dist: pre-commit>=3.5.0
+Requires-Dist: triton>=2.0.0; sys_platform == "linux"
 
 <div align="center">
 
@@ -210,20 +197,20 @@ Deploy models anywhere:
 
 ### Installation
 
+#### From PyPI (recommended for all users)
+
 ```bash
-# Install from PyPI (recommended)
 pip install wavedl
-
-# Or install with all extras (ONNX export, HPO, dev tools)
-pip install wavedl[all]
 ```
 
+This installs everything you need: training, inference, HPO, ONNX export, and dev tools.
+
 #### From Source (for development)
 
 ```bash
 git clone https://github.com/ductho-le/WaveDL.git
 cd WaveDL
-pip install -e ".[dev]"
+pip install -e .
 ```
 
 > [!NOTE]
@@ -359,41 +346,47 @@ WaveDL handles everything else: training loop, logging, checkpoints, multi-GPU,
 ```
 WaveDL/
 ├── src/
-│   └── wavedl/                # Main package (namespaced)
-│       ├── __init__.py        # Package init with __version__
-│       ├── train.py           # Training entry point
-│       ├── test.py            # Testing & inference script
-│       ├── hpo.py             # Hyperparameter optimization
-│       ├── hpc.py             # HPC distributed training launcher
+│   └── wavedl/                   # Main package (namespaced)
+│       ├── __init__.py           # Package init with __version__
+│       ├── train.py              # Training entry point
+│       ├── test.py               # Testing & inference script
+│       ├── hpo.py                # Hyperparameter optimization
+│       ├── hpc.py                # HPC distributed training launcher
 │       │
-│       ├── models/            # Model architectures
-│       │   ├── registry.py    # Model factory (@register_model)
-│       │   ├── base.py        # Abstract base class
-│       │   ├── cnn.py         # Baseline CNN
-│       │   ├── resnet.py      # ResNet-18/34/50 (1D/2D/3D)
-│       │   ├── efficientnet.py# EfficientNet-B0/B1/B2
-│       │   ├── vit.py         # Vision Transformer (1D/2D)
-│       │   ├── convnext.py    # ConvNeXt (1D/2D/3D)
-│       │   ├── densenet.py    # DenseNet-121/169 (1D/2D/3D)
-│       │   └── unet.py        # U-Net / U-Net Regression
+│       ├── models/               # Model architectures (38 variants)
+│       │   ├── registry.py       # Model factory (@register_model)
+│       │   ├── base.py           # Abstract base class
+│       │   ├── cnn.py            # Baseline CNN (1D/2D/3D)
+│       │   ├── resnet.py         # ResNet-18/34/50 (1D/2D/3D)
+│       │   ├── resnet3d.py       # ResNet3D-18, MC3-18 (3D only)
+│       │   ├── tcn.py            # TCN (1D only)
+│       │   ├── efficientnet.py   # EfficientNet-B0/B1/B2 (2D)
+│       │   ├── efficientnetv2.py # EfficientNetV2-S/M/L (2D)
+│       │   ├── mobilenetv3.py    # MobileNetV3-Small/Large (2D)
+│       │   ├── regnet.py         # RegNetY variants (2D)
+│       │   ├── swin.py           # Swin Transformer (2D)
+│       │   ├── vit.py            # Vision Transformer (1D/2D)
+│       │   ├── convnext.py       # ConvNeXt (1D/2D/3D)
+│       │   ├── densenet.py       # DenseNet-121/169 (1D/2D/3D)
+│       │   └── unet.py           # U-Net Regression
 │       │
-│       └── utils/             # Utilities
-│           ├── data.py        # Memory-mapped data pipeline
-│           ├── metrics.py     # R², Pearson, visualization
-│           ├── distributed.py # DDP synchronization
-│           ├── losses.py      # Loss function factory
-│           ├── optimizers.py  # Optimizer factory
-│           ├── schedulers.py  # LR scheduler factory
-│           └── config.py      # YAML configuration support
+│       └── utils/                # Utilities
+│           ├── data.py           # Memory-mapped data pipeline
+│           ├── metrics.py        # R², Pearson, visualization
+│           ├── distributed.py    # DDP synchronization
+│           ├── losses.py         # Loss function factory
+│           ├── optimizers.py     # Optimizer factory
+│           ├── schedulers.py     # LR scheduler factory
+│           └── config.py         # YAML configuration support
 │
-├── configs/                   # YAML config templates
-├── examples/                  # Ready-to-run examples
-├── notebooks/                 # Jupyter notebooks
-├── unit_tests/                # Pytest test suite (422 tests)
+├── configs/                      # YAML config templates
+├── examples/                     # Ready-to-run examples
+├── notebooks/                    # Jupyter notebooks
+├── unit_tests/                   # Pytest test suite (704 tests)
 │
-├── pyproject.toml             # Package config, dependencies
-├── CHANGELOG.md               # Version history
-└── CITATION.cff               # Citation metadata
+├── pyproject.toml                # Package config, dependencies
+├── CHANGELOG.md                  # Version history
+└── CITATION.cff                  # Citation metadata
 ```
 ---
 
@@ -412,33 +405,63 @@ WaveDL/
 > ```
 
 <details>
-<summary><b>Available Models</b> — 21 pre-built architectures</summary>
-
-| Model | Best For | Params (2D) | Dimensionality |
-|-------|----------|-------------|----------------|
-| `cnn` | Baseline, lightweight | 1.7M | 1D/2D/3D |
-| `resnet18` | Fast training, smaller datasets | 11.4M | 1D/2D/3D |
-| `resnet34` | Balanced performance | 21.5M | 1D/2D/3D |
-| `resnet50` | High capacity, complex patterns | 24.6M | 1D/2D/3D |
-| `resnet18_pretrained` | **Transfer learning** ⭐ | 11.4M | 2D only |
-| `resnet50_pretrained` | **Transfer learning** ⭐ | 24.6M | 2D only |
-| `efficientnet_b0` | Efficient, **pretrained** ⭐ | 4.7M | 2D only |
-| `efficientnet_b1` | Efficient, **pretrained** ⭐ | 7.2M | 2D only |
-| `efficientnet_b2` | Efficient, **pretrained** ⭐ | 8.4M | 2D only |
-| `vit_tiny` | Transformer, small datasets | 5.4M | 1D/2D |
-| `vit_small` | Transformer, balanced | 21.5M | 1D/2D |
-| `vit_base` | Transformer, high capacity | 85.5M | 1D/2D |
-| `convnext_tiny` | Modern CNN, transformer-inspired | 28.2M | 1D/2D/3D |
-| `convnext_tiny_pretrained` | **Transfer learning** ⭐ | 28.2M | 2D only |
-| `convnext_small` | Modern CNN, balanced | 49.8M | 1D/2D/3D |
-| `convnext_base` | Modern CNN, high capacity | 88.1M | 1D/2D/3D |
-| `densenet121` | Feature reuse, small data | 7.5M | 1D/2D/3D |
-| `densenet121_pretrained` | **Transfer learning** ⭐ | 7.5M | 2D only |
-| `densenet169` | Deeper DenseNet | 13.3M | 1D/2D/3D |
-| `unet` | Spatial output (velocity fields) | 31.0M | 1D/2D/3D |
-| `unet_regression` | Multi-scale features for regression | 31.1M | 1D/2D/3D |
-
-> ⭐ **Pretrained models** use ImageNet weights for transfer learning.
+<summary><b>Available Models</b> — 38 architectures</summary>
+
+| Model | Params | Dim |
+|-------|--------|-----|
+| **CNN** — Convolutional Neural Network |||
+| `cnn` | 1.7M | 1D/2D/3D |
+| **ResNet** — Residual Network |||
+| `resnet18` | 11.4M | 1D/2D/3D |
+| `resnet34` | 21.5M | 1D/2D/3D |
+| `resnet50` | 24.6M | 1D/2D/3D |
+| `resnet18_pretrained` ⭐ | 11.4M | 2D |
+| `resnet50_pretrained` ⭐ | 24.6M | 2D |
+| **ResNet3D** — 3D Residual Network |||
+| `resnet3d_18` | 33.6M | 3D |
+| `mc3_18` — Mixed Convolution 3D | 11.9M | 3D |
+| **TCN** — Temporal Convolutional Network |||
+| `tcn_small` | 1.0M | 1D |
+| `tcn` | 7.0M | 1D |
+| `tcn_large` | 10.2M | 1D |
+| **EfficientNet** — Efficient Neural Network |||
+| `efficientnet_b0` ⭐ | 4.7M | 2D |
+| `efficientnet_b1` ⭐ | 7.2M | 2D |
+| `efficientnet_b2` ⭐ | 8.4M | 2D |
+| **EfficientNetV2** — Efficient Neural Network V2 |||
+| `efficientnet_v2_s` ⭐ | 21.0M | 2D |
+| `efficientnet_v2_m` ⭐ | 53.6M | 2D |
+| `efficientnet_v2_l` ⭐ | 118.0M | 2D |
+| **MobileNetV3** — Mobile Neural Network V3 |||
+| `mobilenet_v3_small` ⭐ | 1.1M | 2D |
+| `mobilenet_v3_large` ⭐ | 3.2M | 2D |
+| **RegNet** — Regularized Network |||
+| `regnet_y_400mf` ⭐ | 4.0M | 2D |
+| `regnet_y_800mf` ⭐ | 5.8M | 2D |
+| `regnet_y_1_6gf` ⭐ | 10.5M | 2D |
+| `regnet_y_3_2gf` ⭐ | 18.3M | 2D |
+| `regnet_y_8gf` ⭐ | 37.9M | 2D |
+| **Swin** — Shifted Window Transformer |||
+| `swin_t` ⭐ | 28.0M | 2D |
+| `swin_s` ⭐ | 49.4M | 2D |
+| `swin_b` ⭐ | 87.4M | 2D |
+| **ConvNeXt** — Convolutional Next |||
+| `convnext_tiny` | 28.2M | 1D/2D/3D |
+| `convnext_small` | 49.8M | 1D/2D/3D |
+| `convnext_base` | 88.1M | 1D/2D/3D |
+| `convnext_tiny_pretrained` ⭐ | 28.2M | 2D |
+| **DenseNet** — Densely Connected Network |||
+| `densenet121` | 7.5M | 1D/2D/3D |
+| `densenet169` | 13.3M | 1D/2D/3D |
+| `densenet121_pretrained` ⭐ | 7.5M | 2D |
+| **ViT** — Vision Transformer |||
+| `vit_tiny` | 5.5M | 1D/2D |
+| `vit_small` | 21.6M | 1D/2D |
+| `vit_base` | 85.6M | 1D/2D |
+| **U-Net** — U-shaped Network |||
+| `unet_regression` | 31.1M | 1D/2D/3D |
+
+> ⭐ = Pretrained on ImageNet. Recommended for smaller datasets.
 
 </details>
 
@@ -653,12 +676,7 @@ seed: 2025
 
 Automatically find the best training configuration using [Optuna](https://optuna.org/).
 
-**Step 1: Install**
-```bash
-pip install -e ".[hpo]"
-```
-
-**Step 2: Run HPO**
+**Run HPO:**
 
 You specify which models to search and how many trials to run:
 ```bash
@@ -715,7 +733,7 @@ accelerate launch -m wavedl.train --data_path train.npz --model cnn --lr 3.2e-4
 | `--output` | `hpo_results.json` | Output file |
 
 > [!TIP]
-> See [Available Models](#available-models) for all 21 architectures you can search.
+> See [Available Models](#available-models) for all 38 architectures you can search.
 
 </details>
 

wavedl-1.4.0.dist-info/RECORD

@@ -0,0 +1,37 @@
+wavedl/__init__.py,sha256=rGJqm3RDNQt8I9vv57urpUSDeLrmKFFYFFbQj7JC85k,1177
+wavedl/hpc.py,sha256=87oGGDj1lRJaM01eInHPJX3L-G5aexC3Y06376BdVhg,7321
+wavedl/hpo.py,sha256=aZoa_Oto_anZpIhz-YM6kN8KxQXTolUvDEyg3NXwBrY,11542
+wavedl/test.py,sha256=jZmRJaivYYTMMTaccCi0yQjHOfp0a9YWR1wAPeKFH-k,36246
+wavedl/train.py,sha256=LtPJqULHZU3tPbJNlSwK4GlLxQB1Df2yQfWiEHMn54o,44945
+wavedl/models/__init__.py,sha256=lfSohEnAUztO14nuwayMJhPjpgySzRN3jGiyAUuBmAU,3206
+wavedl/models/_template.py,sha256=TYQSeH6NIHf9iqJ4n6PK0s7xsjitxFsu-n1EdWL24AA,4822
+wavedl/models/base.py,sha256=atM7NJQhtgpA-NauLBVUkXTd8XccjGPBqg3qxX05SRo,5289
+wavedl/models/cnn.py,sha256=rn2Xmup0w_ll6wuAnYclSeIVazoSUrUGPY-9XnhA1gE,8341
+wavedl/models/convnext.py,sha256=5zELY0ztMB6FxJB9uBurloT7JBdxLXezmrNRzLQjrI0,12846
+wavedl/models/densenet.py,sha256=LzNbQOvtcJJ4SVf-XvIlXGNUgVS2SXl-MMPbr8lcYrA,12995
+wavedl/models/efficientnet.py,sha256=0DHBgEGaOucevtmO1KPUTb5bCdJRg-Gzfpu9EuaylGQ,7456
+wavedl/models/efficientnetv2.py,sha256=rP8y1ZAWyNyi0PXGPXg-4HjgzoELZ-CjMFgr8WnSXeg,10244
+wavedl/models/mobilenetv3.py,sha256=h3f6TiNSyHRH9Qidce7dCGTbdEWYfYF5kbU-TFoTg0U,9490
+wavedl/models/registry.py,sha256=InYAXX2xbRvsFDFnYUPCptJh0F9lHlFPN77A9kqHRT0,2980
+wavedl/models/regnet.py,sha256=Yf9gAoDLv0j4uEuoKC822gizHNh59LCbvFCMP11Q1C0,13116
+wavedl/models/resnet.py,sha256=8DNGIrH5pK8pjEE9BSyBqIc_pkFS_qaYggx-stjTF5k,16961
+wavedl/models/resnet3d.py,sha256=C7CL4XeSnRlIBuwf5Ei-z183uzIBObrXfkM9Iwuc5e0,8746
+wavedl/models/swin.py,sha256=p-okfq3Qm4_neJTxCcMzoHoVzC0BHW3BMnbpr_Ri2U0,13224
+wavedl/models/tcn.py,sha256=RtY13QpFHqz72b4ultv2lStCIDxfvjySVe5JaTx_GaM,12601
+wavedl/models/unet.py,sha256=LqIXhasdBygwP7SZNNmiW1bHMPaJTVBpaeHtPgEHkdU,7790
+wavedl/models/vit.py,sha256=0C3GZk11VsYFTl14d86Wtl1Zk1T5rYJjvkaEfEN4N3k,11100
+wavedl/utils/__init__.py,sha256=YMgzuwndjr64kt9k0_6_9PMJYTVdiaH5veSMff_ZycA,3051
+wavedl/utils/config.py,sha256=E0_m5aQ1OdwEwzZysSwc5v905P4g3SDprObFAeVIj9g,8107
+wavedl/utils/cross_validation.py,sha256=117ac9KDzaIaqhtP8ZRs15Xpqmq5fLpX2-vqkNvtMaU,17487
+wavedl/utils/data.py,sha256=9LrB9MC6jRZzbRSc9xiGzJWoh8FahwP_68REqBAT3Os,44131
+wavedl/utils/distributed.py,sha256=7wQ3mRjkp_xjPSxDWMnBf5dSkAGUaTzntxbz0BhC5v0,4145
+wavedl/utils/losses.py,sha256=5762M-TBC_hz6uyj1NPbU1vZeFOJQq7fR3-j7OygJRo,7254
+wavedl/utils/metrics.py,sha256=mkCpqZwl_XUpNvA5Ekjf7y-HqApafR7eR6EuA8cBdM8,37287
+wavedl/utils/optimizers.py,sha256=PyIkJ_hRhFi_Fio81Gy5YQNhcME0JUUEl8OTSyu-0RA,6323
+wavedl/utils/schedulers.py,sha256=e6Sf0yj8VOqkdwkUHLMyUfGfHKTX4NMr-zfgxWqCTYI,7659
+wavedl-1.4.0.dist-info/LICENSE,sha256=cEUCvcvH-9BT9Y-CNGY__PwWONCKu9zsoIqWA-NeHJ4,1066
+wavedl-1.4.0.dist-info/METADATA,sha256=NDjDovaoOPWEno7rQVIId5_kOSyYrTA0nQgkyOfbdXI,39286
+wavedl-1.4.0.dist-info/WHEEL,sha256=beeZ86-EfXScwlR_HKu4SllMC9wUEj_8Z_4FJ3egI2w,91
+wavedl-1.4.0.dist-info/entry_points.txt,sha256=f1RNDkXFZwBzrBzTMFocJ6xhfTvTmaEDTi5YyDEUaF8,140
+wavedl-1.4.0.dist-info/top_level.txt,sha256=ccneUt3D5Qzbh3bsBSSrq9bqrhGiogcWKY24ZC4Q6Xw,7
+wavedl-1.4.0.dist-info/RECORD,,

wavedl-1.3.0.dist-info/RECORD

@@ -1,31 +0,0 @@
-wavedl/__init__.py,sha256=tyDZPCdra3-5TMWwIE4AvoDTkdHq-KZ_vp4lPgCfuX8,1177
-wavedl/hpc.py,sha256=OaiGo0Q_ylu6tCEZSnMZ9ohk3nWcqbnwNMXrbZgikF0,7325
-wavedl/hpo.py,sha256=aZoa_Oto_anZpIhz-YM6kN8KxQXTolUvDEyg3NXwBrY,11542
-wavedl/test.py,sha256=jZmRJaivYYTMMTaccCi0yQjHOfp0a9YWR1wAPeKFH-k,36246
-wavedl/train.py,sha256=eIEN53n7ZrJXfcR_TUcL0cdDFzU8BY2L6XbxqGiUnbA,43887
-wavedl/models/__init__.py,sha256=AbsFkRNlsiWv4sJ-kLPdwjA2FS_cSp_TB3CV8884uUE,2219
-wavedl/models/_template.py,sha256=O7SfL3Ef7eDXGmcOXPD0c82o_t3K4ybgJwpSEDsZNEg,4837
-wavedl/models/base.py,sha256=cql0wv8i1sMaVttXOSdBBTPfa2s2sLH5LyAsfKJdXX8,5304
-wavedl/models/cnn.py,sha256=2FFQetQaCJqeeku6glXbOQ3KJw5VvSTu9-u9cpygVk8,8356
-wavedl/models/convnext.py,sha256=zh-x5NFcZrcRv3bi55p-VKWHLYe-v1nvPcMp9xPizLk,12747
-wavedl/models/densenet.py,sha256=q9qrgnacMQ1GDGGPks0jx-C3DRjacnTV8BQ-iw6BTFY,12864
-wavedl/models/efficientnet.py,sha256=irxab-yt3z89tMTf1x6odR2IqgpMrMM44Wiu3n6-IEs,7285
-wavedl/models/registry.py,sha256=p5Eof3T6cwHggcEM-xzeBoKMbpuNyRmOJIvqMhzHvJA,2995
-wavedl/models/resnet.py,sha256=sT4S_Rx56dqLN5zEPbBKeJet1dvr49IWhnBSjiVfcQs,16777
-wavedl/models/unet.py,sha256=i3DFpeJmvdzNiBSqi4ecjLbC9RZXXbNJ_ZNMr2c3I6I,10019
-wavedl/models/vit.py,sha256=iyJ8FQ1DOAgBhaVIUGGQEP2L37wZ28JeHKkJ1tmgj9w,10898
-wavedl/utils/__init__.py,sha256=YMgzuwndjr64kt9k0_6_9PMJYTVdiaH5veSMff_ZycA,3051
-wavedl/utils/config.py,sha256=E0_m5aQ1OdwEwzZysSwc5v905P4g3SDprObFAeVIj9g,8107
-wavedl/utils/cross_validation.py,sha256=117ac9KDzaIaqhtP8ZRs15Xpqmq5fLpX2-vqkNvtMaU,17487
-wavedl/utils/data.py,sha256=9LrB9MC6jRZzbRSc9xiGzJWoh8FahwP_68REqBAT3Os,44131
-wavedl/utils/distributed.py,sha256=7wQ3mRjkp_xjPSxDWMnBf5dSkAGUaTzntxbz0BhC5v0,4145
-wavedl/utils/losses.py,sha256=5762M-TBC_hz6uyj1NPbU1vZeFOJQq7fR3-j7OygJRo,7254
-wavedl/utils/metrics.py,sha256=mkCpqZwl_XUpNvA5Ekjf7y-HqApafR7eR6EuA8cBdM8,37287
-wavedl/utils/optimizers.py,sha256=PyIkJ_hRhFi_Fio81Gy5YQNhcME0JUUEl8OTSyu-0RA,6323
-wavedl/utils/schedulers.py,sha256=e6Sf0yj8VOqkdwkUHLMyUfGfHKTX4NMr-zfgxWqCTYI,7659
-wavedl-1.3.0.dist-info/LICENSE,sha256=cEUCvcvH-9BT9Y-CNGY__PwWONCKu9zsoIqWA-NeHJ4,1066
-wavedl-1.3.0.dist-info/METADATA,sha256=FiNI2u9YCcVZUolFgKg9GzNc3i5MC1M_JYR171qvKN8,38922
-wavedl-1.3.0.dist-info/WHEEL,sha256=beeZ86-EfXScwlR_HKu4SllMC9wUEj_8Z_4FJ3egI2w,91
-wavedl-1.3.0.dist-info/entry_points.txt,sha256=f1RNDkXFZwBzrBzTMFocJ6xhfTvTmaEDTi5YyDEUaF8,140
-wavedl-1.3.0.dist-info/top_level.txt,sha256=ccneUt3D5Qzbh3bsBSSrq9bqrhGiogcWKY24ZC4Q6Xw,7
-wavedl-1.3.0.dist-info/RECORD,,