python - 我有一个 pytorch 图像分类器训练，我想暂停训练并在程序暂停时保存权重。我可以这样做吗？

Question

我正在训练一个已经训练了几天的分类器，但我的问题是我没有在整个训练过程中编写代码来保存 .pt 检查点，所以我最终只会得到一个权重文件当程序完成其所有时期时。有没有办法立即暂停训练（PAUSE BREAK）并保存模型的权重？

标签： pythonpytorch

Answer 1

不幸的是，PyTorch 目前没有用于此的原生 API。对于当前的工作，您可以使用像PyDev或Pycharm这样的 IDE将调试器附加到正在运行的进程，并在代码中的某处设置断点并提取权重和偏差。

对于未来的工作，您总是可以在 epochs 循环中创建检查点并将学习的模型保存在那里。这个链接会有所帮助。