聊天机器人教程 · Pytorch 中文文档

[apachecn]

聊天机器人教程 · Pytorch 中文文档

[jiangzhonglian]

如果您不配置此代码，则在从4000_checkpoint.tar迭代更新时将收到错误：

encoder_optimizer.step()  

错误信息

exp_avg.mul_(beta1).add_(1 - beta1, grad)
RuntimeError: expected backend CPU and dtype Float but got backend CUDA and dtype Float

解决方案: pytorch/pytorch#2830

model = Model()
model.load_state_dict(checkpoint['model'])
model.cuda()
optimizer = optim.Adam(model.parameters())
optimizer.load_state_dict(checkpoint['optimizer'])
for state in optimizer.state.values():
    for k, v in state.items():
        if isinstance(v, torch.Tensor):
            state[k] = v.cuda()

已经被官方收录： pytorch/tutorials#577