PyTorch implementation for:
- LeNet5 on MINIST dataset
- LeNet5 and ResNet18 on CIFAR10 dataset
- LeNet5 and ResNet34 on trashnet dataset
- python3
- pytorch
- torchvision
- MINIST
python LeNet5_for_MINIST.py
- CIFAR10
python LeNet5_for_CIFAR10.py
python ResNet18_for_CIFAR10.py
- trashnet
python main.py