Commit e64deb97 authored by Cresson Remi's avatar Cresson Remi

ENH: explicit error message when inputs have not the same size

parent 57045278
......@@ -82,7 +82,8 @@ TensorflowMultisourceModelLearningBase<TInputImage>
}
else if (m_NumberOfSamples != currNumberOfSamples)
{
itkGenericExceptionMacro("Previous batch size is " << m_NumberOfSamples
itkGenericExceptionMacro("Batch size of input " << (i-1)
<< " was " << m_NumberOfSamples
<< " but input " << i
<< " has a batch size of " << currNumberOfSamples );
}
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment