change_detection.pytorch icon indicating copy to clipboard operation
change_detection.pytorch copied to clipboard

Published 20 hours ago • verified publisher icon likyoo

RuntimeError: Attempted to set the storage of a tensor on device "cpu" to a storage on different device "cuda:0". This is no longer allowed; the devices must match.

Open liujiaoshou17 opened this issue 1 year ago • 1 comments

RuntimeError: Attempted to set the storage of a tensor on device "cpu" to a storage on different device "cuda:0". This is no longer allowed; the devices must match. 使用预训练模型为imagenet时会报错，其他预训练模型就可以

liujiaoshou17 avatar May 26 '23 07:05 liujiaoshou17