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

Question

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

Answer 1

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

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