has anyone experienced an issue when the output shape of the prediction is changed after one epoch?
Context: I am training a neural network model using multi_gpu mode
