notebooks icon indicating copy to clipboard operation
notebooks copied to clipboard

Published 20 hours ago • verified publisher icon huggingface

RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same or input should be a MKLDNN tensor and weight is a dense tensor

Open Dexter-Waston opened this issue 2 years ago • 1 comments
trafficstars

when it comes to the step “trainer.train()” on the colab, it returns "RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same or input should be a MKLDNN tensor and weight is a dense tensor". I'm wondering how to solve it.

Dexter-Waston avatar Dec 03 '22 07:12 Dexter-Waston