分类：model.to(device)

2021
07-12

pytorch中的model=model.to(device)使用说明

这代表将模型加载到指定设备上。其中，device=torch.device("cpu")代表的使用cpu，而device=torch.device("cuda")则代表的使用GPU。当我们指定了设备之后，就需要将模型加载到相应设备中，此时需要使用model=model.to(device)，将模型加载到相应的设备中。将由GPU保存的模型加载到CPU上。将torch.load()函数中的map_location参数设置为torch.device('cpu')device=torch.device('cpu')model=TheModelClass(*args,**kwargs)mode... 继续阅读 >