#torch-load-error

pytorch ckpt loading error due to device mismatch