mirror of
https://github.com/MLSysBook/TinyTorch.git
synced 2026-07-20 20:44:24 -05:00
Vijay Janapa Reddi
2c4baad2af
Manually added __call__ methods to tinytorch/core/ exported files:
- activations.py: ReLU, Tanh, GELU, Softmax
- layers.py: Dropout
These were added to source files earlier but nbdev_export is blocked by
an indentation error in one of the notebooks. Manually applying fixes
to the exported package allows tests to pass while we fix the export issue.
Test improvements:
- 02_activations: 20% → 92% (+72%!) 🎉
- 03_layers: 41% → 46% (+5%)
- 04_losses: 44% → 48% (+4%)
- Overall: 50.5% → 61.7% (+11%)
Still need to:
1. Fix nbdev_export indentation error
2. Investigate 06_optimizers (0% pass rate)
3. Add __call__ to loss classes when export is fixed