Checking.py
import torch
import torch.nn as nn
import torch.optim as optim
import torch.nn.functional as F
import torchvision
import torchvision.transforms as transforms
import numpy as np
from torch.utils.data.dataset import Dataset
ckpt_path=’/home/students/soumyajit_po/hsd_cnn_code/SOUMYA/checkpoint/vggCif    ar10TEST.pth’
model = CifarVggnet(*args, **kwargs)
model.load_state_dict(torch.load(ckpt_path))
print(model)
I am using Checking.py to load the model saved in ckpt_path.