@ptrblck @chhaya_kumar_das But how do you explain that the error occurs after several epochs? All data points have been processed already at that point at least once without illegal memory error, right? I’m having this error too.