保存
state = {'net': model.state_dict(), 'optimizer':optimizer.state_dict(), 'epoch': epoch}
torch.save(state, 'dir.pth')
加载
checkpoint = torch.load(dir)
model.load_state_dict(checkpoint['net'])
optimizer.load_state_dict(checkpoint['optimizer'])
start_epoch = checkpoint['epoch'] + 1
版权声明：本文为weixin_42764932原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接和本声明。