标签：input=input.to(device)

[mcj]pytorch中model=model.to(device)用法

这代表将模型加载到指定设备上。其中，device=torch.device("cpu")代表的使用cpu，而device=torch.device("cuda")则代表的使用GPU。当我们指定了设备之后，就需要将模型加载到相应设备中，此...