diff --git a/mace/ops/deconv_2d_benchmark.cc b/mace/ops/deconv_2d_benchmark.cc index 4305b57504d3dde871b48e07e4b3463f1e88d405..af8d3b87e853b8330df962bf4841f9fbdff3c296 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 ecdc70095c09caca93694691fb0eeb2ca8e1550a..fd1210b6786a0009ab0d814a6ab3ee0dee9e85c5 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