TinyTorch

mirror of https://github.com/MLSysBook/TinyTorch.git synced 2026-06-03 22:51:45 -05:00

Files

Vijay Janapa Reddi 04a387014c Add test timeouts and local test data for data module

- Added pytest-timeout configuration with 5-minute timeout for all tests
- Added timeout handling to test command with proper error messages
- Created small local test dataset (50 train + 20 test samples) that mimics CIFAR-10 structure
- Updated data module tests to use local test data instead of downloading CIFAR-10
- Tests now run much faster (~0.1s vs ~30s) and don't require internet connection
- Added TestCIFAR10Dataset class that loads from local pickle files
- All test functionality preserved but using local data for speed and reliability

2025-07-11 18:50:05 -04:00

activations

Implements activation functions

2025-07-11 15:25:27 -04:00

cnn

Implements 2D Convolutional Neural Network module

2025-07-11 15:21:58 -04:00

data

Add test timeouts and local test data for data module

2025-07-11 18:50:05 -04:00

layers

Implements core neural network layers