Skip to content

Machine Learning

[PyTorch] Tutorial(5) How to train a model to classify CIFAR-10 database

Today we challenged the classifiers of different data sets again. This time, CIFAR-19 is a more difficult problem than MNIST handwriting recognition. In addition to the size of the picture becoming 32×32, CIFAR-10 is no longer a pure grayscale value, but a picture with the three primary colors of RGB.

Read More »[PyTorch] Tutorial(5) How to train a model to classify CIFAR-10 database

Easy way to remove CUDA

Cryptocurrency mining equipment, isometric ethereum digital currency extract, blockchain system dark blue vector illustration

Introduction

If you are familiar with the deep learning and its environment, you may often hear “CUDA“. Before we starting to train our model, maybe we need to install GPU driver, CUDA and cuDNN toolkits. They can make our training have the best effect.

Read More »Easy way to remove CUDA