即 创建了device这个(GPU/CPU) 设备,然后将model(data)分配到device上,使用device对model(data)进行训练学习。
torch.device()
最新推荐文章于 2024-03-28 07:42:00 发布
即 创建了device这个(GPU/CPU) 设备,然后将model(data)分配到device上,使用device对model(data)进行训练学习。