tk04/Digit_Prediction_GPU
Made a model to predict numerical digits with the MNIST dataset using pytorch and torchvision. The model was trained on a GPU and perfomed very well: 97% accuracy.
Jupyter Notebook
Made a model to predict numerical digits with the MNIST dataset using pytorch and torchvision. The model was trained on a GPU and perfomed very well: 97% accuracy.
Jupyter Notebook