提交 ed1d3222 authored 作者: affanv14's avatar affanv14

update message for grouped convolutions

上级 2c6adccc
...@@ -440,7 +440,7 @@ PyGpuArrayObject* corrMM(PyGpuArrayObject *const bottom, ...@@ -440,7 +440,7 @@ PyGpuArrayObject* corrMM(PyGpuArrayObject *const bottom,
" weight shape: %ld %ld %ld %ld\n" " weight shape: %ld %ld %ld %ld\n"
" top shape: %ld %ld %ld %ld (expected %ld %ld %ld %ld)\n", " top shape: %ld %ld %ld %ld (expected %ld %ld %ld %ld)\n",
batchSize, nChannels, bottomHeight, bottomWidth, batchSize, nChannels, bottomHeight, bottomWidth,
nFilters, nChannels, kH, kW, nFilters, nChannels / numgroups, kH, kW,
PyGpuArray_DIMS(top)[0], PyGpuArray_DIMS(top)[1], PyGpuArray_DIMS(top)[0], PyGpuArray_DIMS(top)[1],
PyGpuArray_DIMS(top)[2], PyGpuArray_DIMS(top)[3], PyGpuArray_DIMS(top)[2], PyGpuArray_DIMS(top)[3],
batchSize, nFilters, topHeight, topWidth); batchSize, nFilters, topHeight, topWidth);
......
...@@ -184,7 +184,7 @@ PyArrayObject* corrMM(PyArrayObject* bottom, ...@@ -184,7 +184,7 @@ PyArrayObject* corrMM(PyArrayObject* bottom,
" weight shape: %%d %%d %%d %%d\n" " weight shape: %%d %%d %%d %%d\n"
" top shape: %%ld %%ld %%ld %%ld (expected %%d %%d %%d %%d)\n", " top shape: %%ld %%ld %%ld %%ld (expected %%d %%d %%d %%d)\n",
batchSize, nChannels, bottomHeight, bottomWidth, batchSize, nChannels, bottomHeight, bottomWidth,
nFilters, nChannels, kH, kW, nFilters, nChannels / numgroups, kH, kW,
PyArray_DIMS(top)[0], PyArray_DIMS(top)[1], PyArray_DIMS(top)[0], PyArray_DIMS(top)[1],
PyArray_DIMS(top)[2], PyArray_DIMS(top)[3], PyArray_DIMS(top)[2], PyArray_DIMS(top)[3],
batchSize, nFilters, topHeight, topWidth); batchSize, nFilters, topHeight, topWidth);
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论