Image Classification Using Logistic Regression in PyTorch with CIFAR-10 Data Set

# Imports
import torch
import torchvision
import torch.nn as nn
import matplotlib.pyplot as plt
import torch.nn.functional as F
import torchvision.transforms as transforms
from torchvision.datasets import CIFAR10
from torch.utils.data import random_split
from torch.utils.data import DataLoader
import matplotlib.pyplot as plt
%matplotlib inline

# Hyperparmeters
batch_size = 128

# Other constants
input_size = 3*32*32
num_classes = 10