Add initial counter state to show resumed training from checkpoints

[gabobert]

This is useful when resuming training from a checkpoint at a specific epoch. The starting epoch can now be given to tqdm and the progress bar of the outer loop shows the correct percentage.