Thank you so much for your reply. Following are my imports and I am trying to implement this code GitHub - tjmoon0104/pytorch-tiny-imagenet: pytorch-tiny-imagenet for ResNet18_64
import torch, os
import torch.nn as nn
import torch.optim as optim
import torchvision.datasets as datasets
import torch.utils.data as data
import torchvision.transforms as transforms
import torchvision.models as models
from train_model import train_model
from test_model import test_model
%matplotlib inline