generated from mathias/template-go-web
32 lines
947 B
Markdown
32 lines
947 B
Markdown
# model
|
|
|
|
Python + PyTorch perimeter. This is the only directory in the project that uses Python.
|
|
|
|
## Contents
|
|
|
|
```
|
|
model/
|
|
train.py TS-JEPA training entry point
|
|
configs/ YAML configs per phase
|
|
phase0-mae.yaml
|
|
phase1-tsjepa.yaml
|
|
tsjepa/ TS-JEPA model implementation (adapted from paper)
|
|
tests/ pytest tests for model components
|
|
requirements.txt pinned Python dependencies
|
|
.venv/ (gitignored — created by `task model:setup`)
|
|
```
|
|
|
|
## Setup
|
|
|
|
```bash
|
|
task model:setup # creates .venv and installs requirements via uv
|
|
```
|
|
|
|
## Dependency policy
|
|
|
|
Every Python dependency must be justified in a comment in `requirements.txt`. Prefer Go implementations for anything outside the training loop. When adding a new dependency, add an entry to DECISIONS.md explaining why a Go alternative wasn't sufficient.
|
|
|
|
## Python version
|
|
|
|
3.12 (pinned in `.python-version`)
|