Hi @ptrblck, here is a runnable colab document 5 with everything stripped but the bare requirements to produce the error. The data loaders contain randomly generated tensors of the appropriate dimension. I hope this helps.