It looks like a bug and we would need a code snippet to reproduce this issue.
Based on your last sentence it seems that random input data works fine, while your training workflow produces this error?
Would it be possible to save the particular input batch, which creates the issue, and upload it?