深度学习pytorch程序中途断了怎样接着训练
Posted
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了深度学习pytorch程序中途断了怎样接着训练相关的知识,希望对你有一定的参考价值。
参考技术A 看你有没有保存模型和参数,如果保存了,加载进来继续训练即可。用torch.save保存模型,torch.load加载模型。也可以仅保存参数。
以上是关于深度学习pytorch程序中途断了怎样接着训练的主要内容,如果未能解决你的问题,请参考以下文章
神经网络学习小记录70——Pytorch 使用Google Colab进行深度学习
3. 使用PyTorch深度学习库训练第一个卷积神经网络CNN