tiny-dnn示例在训练时中断，当使用2个类而不是10个时

Question

我对tiny-dnn很新。这个example适用于我的数据：

net << convolutional_layer(32, 32, 5, 1, 6) << tanh() // 32x32in, conv5x5
    << average_pooling_layer(28, 28, 6, 2) << tanh()  // 28x28in, pool2x2
    << fully_connected_layer(14 * 14 * 6, 120) << tanh()
    << fully_connected_layer(120, 10);

当我尝试将输出减少到我的实际类数2时 - 示例中断。它是工作代码和不工作代码之间的变化 - 10改变为2. 2是类的特殊数字吗？

Answer 1

我找到了理由。我列举了从1到1,2开始的类。有一个断言检查类的数量。类的数量应该大于最后一个类id。 将类ID重命名为0,1而不是1,2后，一切正常。