gpu - 在带有 caffe 的 CNN 中，我可以设置初始 caffemodel 吗？

Question

我使用 caffe 操作过 CNN。但是，系统被强制终止。到目前为止，我有 caffemodel。

我可以从现在开始使用当前的 caffemodel 重新学习吗？

谢谢，

Answer 1

Caffe 支持恢复，如下所述：

我们都经历过断电的时候 [...] 由于我们在训练期间对中间结果进行快照，因此我们将能够从快照中恢复。

这可以通过--snapshot主 caffe 命令行工具的选项获得，例如：

./build/tools/caffe train [...] --snapshot=caffenet_train_10000.solverstate

正如文档caffenet_train_10000.solverstate中解释的那样：

求解器状态快照，存储恢复精确求解器状态所需的所有信息。

特别是，您可以从求解器文档中找到有关如何配置快照的更多精度。