From bd54eb98346974b5c289f7c0f4861e4d373c79bb Mon Sep 17 00:00:00 2001 From: xzl Date: Wed, 19 Jul 2017 23:56:23 +0800 Subject: [PATCH] tiny modify the test --- paddle/function/ConvOpTest.cpp | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/paddle/function/ConvOpTest.cpp b/paddle/function/ConvOpTest.cpp index c96c8d9eeab..f0c45c97b16 100644 --- a/paddle/function/ConvOpTest.cpp +++ b/paddle/function/ConvOpTest.cpp @@ -37,7 +37,7 @@ public: for (size_t inputSize : {7, 14, 54}) { for (size_t filterSize : {1, 3, 5}) { for (size_t inputChannels : {3, 64}) { - for (size_t outputChannels : {3, 64, 128}) { + for (size_t outputChannels : {3, 64}) { for (size_t groups : {1, 3, 64}) { if (inputChannels > outputChannels) break; if (groups != 1 && @@ -135,11 +135,10 @@ public: for (size_t filterHeight : {1, 5}) { for (size_t filterWidth : {3, 7}) { for (size_t inputChannels : {7}) { - for (size_t outputChannels : {7, 32}) { + for (size_t outputChannels : {7}) { for (size_t groups : {1, 7}) { - if (!useGroups && groups != 1 && - (inputChannels != groups || - outputChannels % groups != 0)) + if (groups != 1 && (inputChannels != groups || + outputChannels % groups != 0)) continue; if (!useGroups) groups = 1; -- GitLab