Multiple fixes and support for one than one training epoch in a training step