Всем привет. Кто нить знает как в питорче обучать на
гпу? В последней строчке крашится (Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!), хотя обе переменные на гпу хранятся, сама модель тоже на гпу. Переменная device - device(type='cuda').