This project is an example about how to use libtorch to train a classification model. It include 2 model:
- Handwritten digit recognition
- Dog and Cat classification
The custom dataset is in the dataset folder. The Mnist dataset loads information from a csv filee and the Dog and Cat dataset loads image from a folder. model folder contain Fully Connected Neural Network (FCN) and Convolutional Neural Network (CNN)