Metadata-Version: 2.4
Name: aptt
Version: 0.1.0
Author-email: anton feldmann <anton.feldmann@gmail.com>
License: MIT
License-File: LICENSE
Requires-Python: >=3.11
Requires-Dist: complexpytorch>=0.4
Requires-Dist: loguru>=0.7.3
Requires-Dist: matplotlib>=3.10.3
Requires-Dist: mlflow>=2.22.0
Requires-Dist: motmetrics>=1.4.0
Requires-Dist: numpy>=2.3.0
Requires-Dist: opencv-python-headless>=4.11.0.86
Requires-Dist: plotly>=6.1.2
Requires-Dist: psutil>=7.0.0
Requires-Dist: py-cpuinfo>=9.0.0
Requires-Dist: pytorch-lightning>=2.5.1.post0
Requires-Dist: ray[tune]>=2.47.1
Requires-Dist: toml>=0.10.2
Requires-Dist: torchaudio>=2.6.0
Requires-Dist: torchinfo>=1.8.0
Requires-Dist: torchsummary>=1.5.1
Requires-Dist: torchviz>=0.0.3
Requires-Dist: zarr>=3.0.8
Provides-Extra: cpu
Requires-Dist: torch<3.0.0,>=2.6.0; extra == 'cpu'
Requires-Dist: torchvision<0.22.0,>=0.21.0; extra == 'cpu'
Provides-Extra: cu124
Requires-Dist: torch<3.0.0,>=2.6.0; extra == 'cu124'
Requires-Dist: torchvision<0.22.0,>=0.21.0; extra == 'cu124'
Provides-Extra: dev
Requires-Dist: black>=25.1.0; extra == 'dev'
Requires-Dist: mypy>=1.15.0; extra == 'dev'
Requires-Dist: pytest>=8.3.5; extra == 'dev'
Requires-Dist: ruff>=0.11.11; extra == 'dev'
Requires-Dist: sphinx>=8.2.3; extra == 'dev'
Requires-Dist: types-toml>=0.10.8.20240310; extra == 'dev'
Provides-Extra: doc
Requires-Dist: autodoc-pydantic>=2.2.0; extra == 'doc'
Requires-Dist: furo>=2024.8.6; extra == 'doc'
Requires-Dist: myst-parser>=4.0.1; extra == 'doc'
Requires-Dist: pygraphviz>=1.14; extra == 'doc'
Requires-Dist: recommonmark>=0.7.1; extra == 'doc'
Requires-Dist: sphinx-autodoc-typehints>=3.2.0; extra == 'doc'
Requires-Dist: sphinx>=8.2.3; extra == 'doc'
Description-Content-Type: text/markdown

# APTT – Antons PyTorch Tools

**APTT** (Antons PyTorch Tools) is a modular, extensible deep learning framework designed to streamline training, 
evaluation, and experimentation using [PyTorch Lightning](https://www.pytorchlightning.ai/). It supports a wide range of model architectures, 
loss functions, evaluation metrics, and training utilities—across both vision and audio domains.

## Features

- ✅ Wide range of supported model types (YOLO, ResNet, RNNs, WaveNet, etc.)
- 🧩 Pluggable callbacks (TorchScript export, TensorRT optimization, t-SNE visualization, etc.)
- 🧠 Built-in continual learning and knowledge distillation
- ⚙️ Modular structure (Heads, Losses, Layers, Metrics, Callbacks, etc.)
- 📊 Embedding visualization & analysis tools
- 🗂️ Flexible dataset loaders for audio and image tasks
- 🧪 Unit tests and full documentation with Sphinx

## Project Structure

```bash
aptt/
├── aptt/                  # Core source code (models, callbacks, utils, etc.)
├── tests/                 # Unit tests
├── docs/                  # Sphinx-based documentation
├── README.md              # This file
├── pyproject.toml         # Build system and dependencies
└── LICENSE                # License information
```

---

## Installation
```bash
# Clone the repository
git clone https://github.com/your-user/aptt.git
cd aptt

# (Optional) Create and activate a virtual environment
python -m venv venv
source venv/bin/activate  # or venv\Scripts\activate on Windows

# (For Sphinx Documentation)
apt-get install libgraphviz-dev


# Install dependencies
uv install .
```

## Quick Start Example
```python
from aptt.lightning_base.trainer import APTTTrainer

trainer = APTTTrainer(config_path="config.yaml")
trainer.train()
```

## Documentation
To build the documentation locally:

```bash
cd docs
make html
```

The HTML output will be located in docs/_build/html/index.html.

## License
This project is licensed under the MIT License – see the LICENSE file for details.
