name ‘device‘ is not defined

在训练简单的CCN进行模型选择时，出现name ‘device’ is not defined

发现是没有定义device，因此查看了以下model = model.to(device)这行代码的意义，

https://blog.youkuaiyun.com/weixin_36670529/article/details/104367696博客中描述的很详细**，是说将模型加载到指定设备上**。

其中，device=torch.device(“cpu”)代表的使用cpu，而device=torch.device(“cuda”)则代表的使用GPU。

由于我是在服务器上跑的，用的时GPU，所以在代码前加了一行代码：

device=torch.device(“cuda”)
错误改好。

device = torch.device(“cuda”)
model = model.to(device)