batch number for task incorrect in training loop

[syncrostone]

batch number is incremented once at the beginning of training loop and then once within each loop, or twice when curriculum is being used. fix this by using different batch generators for each purpose

[carolinehaoud]

I'm going to try working on this.