# pytorch_template

> AI-agent-friendly PyTorch research pipeline — one YAML config drives preflight, training, Optuna HPO, and real-time TUI monitoring

- **URL**: https://www.freshcrate.ai/projects/pytorch_template
- **Author**: Axect
- **Category**: AI Agents
- **Latest version**: `v0.3.0` (2026-04-08)
- **License**: MIT
- **Source**: https://github.com/Axect/pytorch_template
- **Homepage**: https://axect.github.io/pytorch_template
- **Language**: Python
- **GitHub**: 10 stars, 1 forks
- **Registry**: github
- **Tags**: `ai-agent`, `claude-code`, `cli`, `deep-learning`, `experiment-management`, `hyperparameter-optimization`, `machine-learning`, `optuna`, `python`

## Description

AI-agent-friendly PyTorch research pipeline — one YAML config drives preflight, training, Optuna HPO, and real-time TUI monitoring

## Recent releases

| Version | Date | Urgency | Changes |
| --- | --- | --- | --- |
| `v0.3.0` | 2026-04-08 | High | ## New Features  - **Dual logging**: `logging: wandb` (default) or `logging: tui` for agent-friendly terminal output - **CSVLoggingCallback** (always active): writes `metrics.csv` every epoch with dynamic column expansion - **TUILoggingCallback**: structured per-epoch terminal output replacing W&B - **LatestModelCallback** (always active): saves `latest_model.pt` every epoch - **Rust TUI monitor** (`tools/monitor/`): real-time loss curve visualization from `metrics.csv` - **Provenance tracking** |
| `v0.2.0` | 2026-03-27 | Medium | ## What's New  ### Pre-flight Check Run 1 batch forward+backward before training to catch config errors in seconds: ```bash python -m cli preflight configs/run_template.yaml --device cuda:0 ``` Detects shape mismatches, NaN/Inf gradients, scheduler param issues, and estimates GPU memory.  ### HPO Analysis After HPO, understand what Optuna found: ```bash python -m cli hpo-report --opt-config configs/my_opt.yaml ``` Shows parameter importance (fANOVA), boundary warnings, and top-K trial comparison |

## Citation

- HTML: https://www.freshcrate.ai/projects/pytorch_template
- Markdown: https://www.freshcrate.ai/projects/pytorch_template.md
- Dependencies JSON: https://www.freshcrate.ai/api/projects/pytorch_template/deps

_Generated by freshcrate.ai. Indexes github releases for AI-agent ecosystem packages._