Cats/Dogs Image Classification using Transfer Learning(ResNet34) in PyTorch

import os
import torch
import torchvision
import zipfile
import torchvision.transforms as T
import torch.nn as nn
import torch.nn.functional as F
import torchvision.models as models
from torchvision.datasets.utils import download_url
from torch.utils.data import random_split
from torchvision.datasets import ImageFolder
from torchvision.transforms import ToTensor
from torch.utils.data import DataLoader


image_size = 64
batch_size = 128

# Dowload the dataset
dataset_url = "http://files.fast.ai/data/dogscats.zip"
download_url(dataset_url, '.')

with zipfile.ZipFile('./dogscats.zip') as zip:
    zip.extractall(path='./data')