From ce431bbff32309f98ab91b4cd3932ee3d6e53f35 Mon Sep 17 00:00:00 2001 From: liutuo Date: Mon, 21 May 2018 15:36:24 +0800 Subject: [PATCH] modify deconv and fc benchmark OC size smaller --- mace/ops/deconv_2d_benchmark.cc | 2 +- mace/ops/fully_connected_benchmark.cc | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/mace/ops/deconv_2d_benchmark.cc b/mace/ops/deconv_2d_benchmark.cc index 4305b575..af8d3b87 100644 --- a/mace/ops/deconv_2d_benchmark.cc +++ b/mace/ops/deconv_2d_benchmark.cc @@ -124,7 +124,7 @@ BM_DECONV_2D(1, 32, 60, 60, 1, 1, 1, 60, 60, VALID, 128); BM_DECONV_2D(1, 128, 60, 60, 3, 3, 1, 62, 62, VALID, 128); BM_DECONV_2D(1, 32, 60, 60, 3, 3, 1, 60, 60, SAME, 32); -BM_DECONV_2D(1, 3, 512, 512, 7, 7, 2, 1023, 1023, SAME, 64); +BM_DECONV_2D(1, 3, 512, 512, 7, 7, 2, 1023, 1023, SAME, 32); BM_DECONV_2D(1, 128, 16, 16, 5, 5, 1, 20, 20, VALID, 32); BM_DECONV_2D(1, 128, 64, 64, 5, 5, 1, 68, 68, VALID, 32); diff --git a/mace/ops/fully_connected_benchmark.cc b/mace/ops/fully_connected_benchmark.cc index ecdc7009..fd1210b6 100644 --- a/mace/ops/fully_connected_benchmark.cc +++ b/mace/ops/fully_connected_benchmark.cc @@ -103,7 +103,7 @@ void FCBenchmark( BM_FC(1, 16, 16, 32, 32); BM_FC(1, 8, 8, 32, 1000); BM_FC(1, 2, 2, 512, 2); -BM_FC(1, 7, 7, 512, 4096); +BM_FC(1, 7, 7, 512, 2048); } // namespace test } // namespace ops -- GitLab