Skip to content

kurokoTurbo/ml_template

Repository files navigation

Vision Template

Local PyTorch training runner with a small FastAPI web UI and import-path based extension points for datasets, models, and losses.

Examples

Run a deterministic synthetic dense-network example:

uv run python -m examples.train_synthetic_dense --epochs 1 --device cpu

Fine-tune a pretrained ResNet-18 on CIFAR-10:

uv run python -m examples.finetune_cifar10_resnet --epochs 1 --device cpu --train-limit 512 --val-limit 128

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors