mirror of
https://github.com/SikongJueluo/Mini-Nav.git
synced 2026-07-12 20:15:31 +08:00
SikongJueluo
4ea567adba
- Add write_training_metrics() in new compressors/training_metrics.py for appending epoch/step/lr/component rows as JSON Lines - Wire --metrics-path and --log-every CLI options into train.py, passing them to the training loop so metrics rows are written every N steps - Accept absolute metrics paths or paths relative to output directory - Add quantization component to loss log alongside existing distill/contrastive - Replace inline torch.device() with get_device() utility - Add test_hash_training_metrics.py covering multi-row JSONL append Infrastructure: - Pin torch 2.7.1 + CUDA 12.8 index for Linux/Windows in pyproject.toml - Add .justfile rsync upload recipe with .stignore exclusion - Exclude **/__marimo__ from rsync in .stignore Dependencies updated: numpy 2.4.5, pandas 3.0.3, black 26.5.0, click 8.4.0, contourpy, etc.
72 lines
1.6 KiB
TOML
72 lines
1.6 KiB
TOML
[project]
|
|
name = "mini-nav"
|
|
version = "0.1.0"
|
|
description = "Add your description here"
|
|
readme = "README.md"
|
|
requires-python = ">=3.10"
|
|
dependencies = [
|
|
"accelerate>=1.13.0",
|
|
"altair>=6.0.0",
|
|
"dash>=3.4.0",
|
|
"dash-ag-grid>=33.3.3",
|
|
"dash-mantine-components>=2.5.1",
|
|
"datasets>=4.5.0",
|
|
"httpx[socks]>=0.28.1",
|
|
"lancedb>=0.30.1",
|
|
"marimo[mcp]>=0.22.0",
|
|
"matplotlib>=3.10.8",
|
|
"polars[database,numpy,pandas,pydantic]>=1.37.1",
|
|
"pyarrow>=23.0.0",
|
|
"pydantic>=2.12.5",
|
|
"pyzmq>=27.1.0",
|
|
"scikit-learn>=1.7.2",
|
|
"transformers>=5.0.0",
|
|
"typer>=0.24.1",
|
|
"torch==2.7.1",
|
|
"torchvision==0.22.1",
|
|
"torchaudio==2.7.1",
|
|
]
|
|
|
|
[dependency-groups]
|
|
dev = [
|
|
"basedpyright>=1.37.4",
|
|
"pip>=26.0.1",
|
|
"cocotb>=2.0.1",
|
|
"pytest>=9.0.2",
|
|
"pytest-mpi>=0.6",
|
|
"python-lsp-server>=1.14.0",
|
|
"ruff>=0.15.7",
|
|
"ty>=0.0.24",
|
|
"websockets>=16.0",
|
|
"cocotb-tools>=0.1.0",
|
|
]
|
|
|
|
[[tool.uv.index]]
|
|
name = "pytorch-cu128"
|
|
url = "https://download.pytorch.org/whl/cu128"
|
|
explicit = true
|
|
|
|
[tool.uv.sources]
|
|
torch = [
|
|
{ index = "pytorch-cu128", marker = "sys_platform == 'linux' or sys_platform == 'win32'" },
|
|
]
|
|
torchvision = [
|
|
{ index = "pytorch-cu128", marker = "sys_platform == 'linux' or sys_platform == 'win32'" },
|
|
]
|
|
torchaudio = [
|
|
{ index = "pytorch-cu128", marker = "sys_platform == 'linux' or sys_platform == 'win32'" },
|
|
]
|
|
|
|
|
|
[tool.ty.environment]
|
|
python = "$UV_PROJECT_ENVIRONMENT"
|
|
root = ["./mini-nav", "./notebooks", "./hw/sim", "./scripts"]
|
|
|
|
[tool.basedpyright]
|
|
include = ["mini-nav", "notebooks"]
|
|
exclude = ["**/node_modules", "**/__pycache__"]
|
|
|
|
|
|
[tool.marimo.runtime]
|
|
pythonpath = ["mini-nav"]
|