Introduction to PyTorch for Image Classification

We will learn how to write code to prepare our dataset and train our model to obtain the final weights and use it during inference in the next step. The flow is shown below:

Data preparation

Data preparation is required to obtain the appropriate format of the data and send it through the model during training. For this purpose, we will learn how to:

• Split the dataset into train, test, and validation

...