SSGC icon indicating copy to clipboard operation
SSGC copied to clipboard

Published 20 hours ago • verified publisher icon allenhaozhu

fix the different device

Open jeongwhanchoi opened this issue 4 years ago • 1 comments

add .item() on validation acc and loss

solve the Runtime error below: RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!

jeongwhanchoi avatar Apr 16 '21 04:04 jeongwhanchoi

add .item() on validation acc and loss

solve the Runtime error below: RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!

Thank you very much for your contribution! I need to review it after UAI 2021 rebuttal period. Sorry for that.

allenhaozhu avatar Apr 16 '21 12:04 allenhaozhu