New backbone

This commit is contained in:
alexiondev
2026-05-08 16:42:35 -04:00
parent 49a2502c76
commit 799aa9fa3d
9 changed files with 69 additions and 22 deletions

View File

@@ -13,6 +13,9 @@ All commands must be run from the project root using the local venv:
# Train the model
.venv/bin/python -m src.models.train --epochs 50 --batch_size 64 --lr 1e-4
# Evaluate a trained model on val and aug_test sets
.venv/bin/python -m src.models.evaluate [--backbone resnet18|resnet34] [--model_path <path>]
# Run inference only (no registry lookup)
.venv/bin/python src/models/inference.py <image_path>