如何培训端到端的CNN?

时间:2016-01-15 18:35:33

标签: caffe

使用caffe作为

  

./ build / tools / caffe train --solver = models / Handmade / solver.prototxt

caffe也进入了阶段:TEST"但我没有测试数据。我只想训练我的训练数据上的参数,所以我还没有使用过"阶段:测试" in" train.prototxt",这会导致错误。我该怎么办?

1 个答案:

答案 0 :(得分:2)

我不知道您是否可以完全省略测试阶段,但可以在不需要单独的测试集的情况下训练您的模型。它也可以防止求解器切换到测试阶段。

重新使用测试阶段的训练数据。您可以通过复制数据层并在测试阶段指定它来完成此操作。

要将计算限制到训练阶段,只需将求解器定义中test_interval的值增加到大于训练集的数字,或者更好,大于max_iter。这可以防止求解器切换到测试阶段。

我觉得训练模型有点奇怪而不想知道如何在一组单独的数据点上做。

相关问题