HarshRangwala:

I tried evaluate(model_name.to(device), val_dl)  but it didn’t work. (I checked and the device is ‘cuda’)


If you want to try things out early, we also have a quantized CUDA test here: pytorch/quantized_resnet_test.py at master · pytorch/pytorch · GitHub 6