模型转化
# GPU模型 -> GPU参数 / CPU模型 -> CPU参数
model.load_state_dict(torch.load('outputs/epoch4.pt'))
# GPU模型 -> CPU 参数
model.load_state_dict(torch.load('outputs/epoch4.pt', map_location=lambda storage, loc: storage.cuda))
# CPU模型 -> GPU参数
model.load_state_dict(torch.load('outputs/epoch4.pt', map_location=lambda storage, loc: storage))