Steps of stage2

[TU2021]

when reproducing your stage2, I find that the max step of stage 2 is 314.
Maybe the your total train epoch in the sh is not 1 but 2?

[TU2021]

And your recall the baichsize diffierence in the paper, but in your sh, the all batchsize are the same with that of deepscaleR, is there any mistake here?

[nick7nlp]

Thank you for pointing out this issue. The batch size was indeed incorrectly written—only stage 1 has a batch size of 128, while all subsequent stages have a batch size of 64.