Files
TinyTorch/modules
Vijay Janapa Reddi 9b245fe5ea Create complete training module with loss functions, metrics, and training loop
- Add training_dev.py with comprehensive educational structure
- Implement MeanSquaredError, CrossEntropyLoss, BinaryCrossEntropyLoss
- Add Accuracy metric with extensible framework
- Create Trainer class for complete training orchestration
- Include comprehensive inline tests for all components
- Add module.yaml with proper dependencies and metadata
- Create detailed README.md with examples and applications
- Add test_training_integration.py with real component integration tests
- Follow TinyTorch NBDev educational pattern with Build → Use → Optimize
- Ready for real-world training workflows with validation and monitoring
2025-07-14 00:42:46 -04:00
..