First version with wrong batch creation - each epoch training on 1 randomly chosen batch multiple times