Vijay Janapa Reddi 857ab221d8 Fix CNN gradient flow with Conv2dBackward and MaxPool2dBackward ...

- Implemented Conv2dBackward class in spatial module for proper gradient computation
- Implemented MaxPool2dBackward to route gradients through max pooling
- Fixed reshape usage in CNN test to preserve autograd graph
- Fixed conv gradient capture timing in test (before zero_grad)
- All 6 CNN parameters now receive gradients and update properly
- CNN learning verification test now passes with 74% accuracy and 63% loss decrease

2025-11-22 17:29:20 -05:00

74 KiB

Raw Permalink Blame History

View Raw