New backbone
This commit is contained in:
@@ -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>
|
||||
|
||||
|
||||
Reference in New Issue
Block a user