File size: 563 Bytes
932bc69 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 | #!/usr/bin/env python3
"""Backward-compatibility shim — use ``speculators train`` instead.
For distributed training::
torchrun --nproc_per_node=<N> -m speculators.train config.yaml
"""
import warnings
from speculators.train.cli import main
from speculators.train.config import TrainConfig
warnings.warn(
"scripts/train.py is deprecated and will be removed in v0.9.0. "
"Use 'speculators train' or 'torchrun -m speculators.train' instead.",
DeprecationWarning,
stacklevel=1,
)
if __name__ == "__main__":
main(TrainConfig.resolve())
|