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

【问题标题】：tiny-dnn example breaks on training, when using 2 classes instead of 10当使用 2 个类而不是 10 个类时，tiny-dnn 示例会中断训练
【发布时间】：2018-03-15 19:41:57
【问题描述】：

我对 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 是类的特殊数字吗？

【问题讨论】：

标签： tiny-dnn

【解决方案1】：

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

【讨论】：