PyTorch 模型迁移：model.to(device)详解

这行代码'model.to(device)'用于将模型移动到指定的设备上进行计算。

'device'是一个表示设备的变量，可以是'cpu'、'cuda'或特定的GPU设备编号（如'cuda:0'）。

通过调用'.to(device)'方法，可以将模型的参数和计算操作移动到指定的设备上。这对于利用GPU进行加速或在不同设备上运行模型非常有用。在训练过程中，输入数据和标签也需要相应地移动到相同的设备上。