I have some code that can vary the batch size as I process the data. I have been trying it with mxnet, and ran into memory corruptions and crashes. I traced the problem to an wrongly sized array, code to reproduce the problem is attached. Ideally mxnet should report the array is of an unexpected size and abort.
testMXNet.zip
I have some code that can vary the batch size as I process the data. I have been trying it with mxnet, and ran into memory corruptions and crashes. I traced the problem to an wrongly sized array, code to reproduce the problem is attached. Ideally mxnet should report the array is of an unexpected size and abort.
testMXNet.zip