torch之数据类型及加载/模型加载在CPU和GPU间的转换

CPU/GPU上加载模型

• 注：以上为原文地址

---

pytorch可以将模型数据加载到CPU或者GPU上进行训练，也可以在两者之间转换，方式如下：

• CPU–>CPU,GPU–>GPU

torch.load('a.pk1')

• GPU–>CPU

torch.load('a.pk1', map_location = lambda storage, loc:storage)

• CPU–>GPU1

torch.load('a.pk1', map_location = lambda = storage, loc: storage.cuda(1))

---

• torch之数据类型以及在GPU/CPU上转换