Hey @name1ess
Can you please try the following script in the same folder?
import torch
from model import ResNet50
model = ResNet50()
path = ‘checkpoint_model_best.pth’
model = torch.load(path)
model = model.to(device)
model.eval()
print(model)
If this works, you can use the model to infer.