Add early support for torchdata.stateful_dataloader.StatefulDataLoader within the Accelerator
#4899
| Job | Run time |
|---|---|
| 2m 0s | |
| 3m 26s | |
| 3m 38s | |
| 2m 50s | |
| 2m 34s | |
| 2m 35s | |
| 2m 32s | |
| 4m 1s | |
| 4m 6s | |
| 9m 22s | |
| 2m 55s | |
| 4m 32s | |
| 4m 58s | |
| 14m 13s | |
| 3m 55s | |
| 3m 30s | |
| 4m 24s | |
| 5m 4s | |
| 4m 46s | |
| 10m 7s | |
| 1h 35m 28s |