提交 ff6ac985 authored 作者: João Victor Risso's avatar João Victor Risso

Change error message in dnn_sptf to use batch size instead of inputs

上级 3eb0c5c5
...@@ -127,7 +127,7 @@ APPLY_SPECIFIC(dnn_sptf)(PyGpuArrayObject * input, ...@@ -127,7 +127,7 @@ APPLY_SPECIFIC(dnn_sptf)(PyGpuArrayObject * input,
if ( PyGpuArray_DIM( input, 0 ) != num_images ) if ( PyGpuArray_DIM( input, 0 ) != num_images )
{ {
PyErr_Format( PyExc_RuntimeError, PyErr_Format( PyExc_RuntimeError,
"GpuDnnTransformer: expected input to have %d inputs, got %d inputs.", "GpuDnnTransformer: expected batch size %d, got %d.",
num_images, PyGpuArray_DIM( input, 0 ) ); num_images, PyGpuArray_DIM( input, 0 ) );
return 1; return 1;
} }
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论