疑问:
softmax 层的实现时, bottom Blob 和 top Blob 的形状 是相同的, 那么 如何控制 softmax 的类别呢
推测:
通常在 softmax 前层会有 一个 全连接层, 会不会 这个全连接的输出 的尺寸中 指明了, 分类的个数, 查看googLeNet 的,train.prototxt,发现果然如此。同时发现,caffe 实现的 common_layer 大都 有一个num_output的参数,用来指定该层的输出 尺寸。
疑问:
softmax 层的实现时, bottom Blob 和 top Blob 的形状 是相同的, 那么 如何控制 softmax 的类别呢
推测:
通常在 softmax 前层会有 一个 全连接层, 会不会 这个全连接的输出 的尺寸中 指明了, 分类的个数, 查看googLeNet 的,train.prototxt,发现果然如此。同时发现,caffe 实现的 common_layer 大都 有一个num_output的参数,用来指定该层的输出 尺寸。