DEA-Net icon indicating copy to clipboard operation
DEA-Net copied to clipboard

训练出现RuntimeError: Tensor for argument #2 'weight' is on CPU, but expected it to be on GPU (while checking arguments for cudnn_batch_norm)的问题

Open ouyangxi0925 opened this issue 9 months ago • 1 comments

Uploading 屏幕截图 2024-05-28 212827.png…

ouyangxi0925 avatar May 28 '24 13:05 ouyangxi0925