在带有caffe的CNN中,我可以设置初始caffemodel吗?

时间:2015-03-10 04:42:27

标签: gpu restart convolution caffe

我使用caffe操作CNN。 但是,系统被迫终止。 到目前为止我有caffemodel。

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

谢谢,

1 个答案:

答案 0 :(得分:0)

Caffe支持按照here解释恢复:

  

我们都经历过电源耗尽的时间[...]由于我们在培训期间快照中间结果,我们将能够从快照中恢复。

这可以通过main caffe命令行工具的--snapshot选项获得,例如:

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

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

  

求解器状态快照,用于存储恢复精确求解器状态的所有必要信息。

特别是,您可以找到有关如何从solver documentation配置快照的更多精确度。