网络训练注意事项

保存的时候记得optimizer也要保存,然后continuously train的时候,除了load模型和参数,也要load optimizer.

torch.save(optimiser.state_dict(), 'optimiser.pth')
optimiser.load_state_dict(torch.load('optimiser.pth'))

参考:https://www.zhihu.com/question/302800925/answer/533858139
 

你可能感兴趣的:(网络训练注意事项)