I have trained my model and now want to test it but I was getting this error while testing
AttributeError: 'OrderedDict' object has no attribute 'cuda'
The error comes after this:
model = torch.load(snapshot_path) model.cuda()
Is there any mistake in my training that it is not able to find the cuda parameter, since I had done the training on GPU.