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

Question

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

Answer 1

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

对于 future 的工作，您始终可以在 epochs 循环内创建检查点并将学习到的模型保存在那里。 This链接会有帮助。