Using your code produces the following error for me:
RuntimeError: module must have its parameters and buffers on device cuda:5 (device_ids[0]) but found one of them on device: cpu
The error occurs in the line outputs = model(original, keypoints).