@Dwight_Foster Thanks a lot. it has worked. I did like:
Load the model and set in eval
model = torchvision.models.resnet18(pretrained=True)
model.fc = torch.nn.Linear(in_features=512, out_features=2)
model.load_state_dict(torch.load(‘defect_classifier.pt’))
model.eval()
Now getting  AssertionError: Torch not compiled with CUDA enabled.
Trying this: ```
pip install torch -f https://download.pytorch.org/whl/torch_stable.html