diff options
Diffstat (limited to 'examples/graph_squeezenet.cpp')
| -rw-r--r-- | examples/graph_squeezenet.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/examples/graph_squeezenet.cpp b/examples/graph_squeezenet.cpp index 195fe0addb..d38cec28c4 100644 --- a/examples/graph_squeezenet.cpp +++ b/examples/graph_squeezenet.cpp @@ -190,7 +190,7 @@ void main_graph_squeezenet(int argc, const char **argv) get_weights_accessor(data_path, "/cnn_data/squeezenet_v1.0_model/conv10_b.npy"), PadStrideInfo(1, 1, 0, 0)) << ActivationLayer(ActivationLayerInfo(ActivationLayerInfo::ActivationFunction::RELU)) - << PoolingLayer(PoolingLayerInfo(PoolingType::AVG, 13, PadStrideInfo(1, 1, 0, 0, DimensionRoundingType::CEIL))) + << PoolingLayer(PoolingLayerInfo(PoolingType::AVG)) << FlattenLayer() << SoftmaxLayer() << Tensor(get_output_accessor(label, 5)); |