提交 bd54eb98 编写于 作者: X xzl

tiny modify the test

上级 1f516fa0
...@@ -37,7 +37,7 @@ public: ...@@ -37,7 +37,7 @@ public:
for (size_t inputSize : {7, 14, 54}) { for (size_t inputSize : {7, 14, 54}) {
for (size_t filterSize : {1, 3, 5}) { for (size_t filterSize : {1, 3, 5}) {
for (size_t inputChannels : {3, 64}) { 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}) { for (size_t groups : {1, 3, 64}) {
if (inputChannels > outputChannels) break; if (inputChannels > outputChannels) break;
if (groups != 1 && if (groups != 1 &&
...@@ -135,10 +135,9 @@ public: ...@@ -135,10 +135,9 @@ public:
for (size_t filterHeight : {1, 5}) { for (size_t filterHeight : {1, 5}) {
for (size_t filterWidth : {3, 7}) { for (size_t filterWidth : {3, 7}) {
for (size_t inputChannels : {7}) { for (size_t inputChannels : {7}) {
for (size_t outputChannels : {7, 32}) { for (size_t outputChannels : {7}) {
for (size_t groups : {1, 7}) { for (size_t groups : {1, 7}) {
if (!useGroups && groups != 1 && if (groups != 1 && (inputChannels != groups ||
(inputChannels != groups ||
outputChannels % groups != 0)) outputChannels % groups != 0))
continue; continue;
if (!useGroups) groups = 1; if (!useGroups) groups = 1;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册