Use step interval in estimated_stepping_batches docs example (#19774

)

docs/source-pytorch/common/trainer.rst

@@ -1229,7 +1229,10 @@ dataloader if hadn't been set up already.
         optimizer = ...
         stepping_batches = self.trainer.estimated_stepping_batches
         scheduler = torch.optim.lr_scheduler.OneCycleLR(optimizer, max_lr=1e-3, total_steps=stepping_batches)
-        return [optimizer], [scheduler]
+        return {
+            "optimizer": optimizer,
+            "lr_scheduler": {"scheduler": scheduler, "interval": "step"},
+        }
 
 state
 *****