mirror of
https://github.com/MLSysBook/TinyTorch.git
synced 2026-06-02 02:53:20 -05:00
1aae7be8fb
Removed complexity from Module 07 (training): - Removed DemoModel and TestModel classes - Unified all tests/demos to use single minimal MockModel - Module now focuses purely on training infrastructure What remains: - Trainer class (the core training orchestrator) - CosineSchedule (learning rate scheduling) - clip_grad_norm (gradient clipping utility) - Training loop mechanics and checkpointing Impact: - Cleaner, more focused module - No distraction from model architecture - Tests training infrastructure, not model building - All tests still pass with simplified mocks The module now teaches exactly what it should: how to train models, not how to build them.