标签：input=input.to(device)

2019-11-21马春杰杰评论(2)赞(0)

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