| 1 | """ |
| 2 | PyTorch training entrypoint for PI0/PI05 with multi-GPU and multi-node (DDP) support. |
| 3 | This script mirrors the behavior of the JAX trainer (`scripts/train.py`) but runs |
| 4 | entirely in PyTorch using the `PI0Pytorch` model and your existing config/data |