Cherry pick `Update `BaseMegatronSampler` for compatibility with PTL's `_BatchProgress` (11016)` into `r2.0.0`
[🤖]: Hi @ashors1 👋,<br><br>we've cherry picked #11016 into `r2.0.0` for you! 🚀<br><br>Please review and approve this cherry pick by your convenience\!