├── .github └── workflows │ └── documentation.yaml ├── .gitignore ├── LICENSE ├── NOTICE ├── README.md ├── docs ├── conf.py └── index.rst ├── examples └── train_classifier.py ├── pyproject.toml ├── setup.cfg ├── setup.py └── torch_dist_utils ├── __init__.py ├── test.py ├── torch_dist_utils.py └── torchrun_slurm.py /.github/workflows/documentation.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crowsonkb/torch-dist-utils/HEAD/.github/workflows/documentation.yaml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crowsonkb/torch-dist-utils/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crowsonkb/torch-dist-utils/HEAD/LICENSE -------------------------------------------------------------------------------- /NOTICE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crowsonkb/torch-dist-utils/HEAD/NOTICE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crowsonkb/torch-dist-utils/HEAD/README.md -------------------------------------------------------------------------------- /docs/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crowsonkb/torch-dist-utils/HEAD/docs/conf.py -------------------------------------------------------------------------------- /docs/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crowsonkb/torch-dist-utils/HEAD/docs/index.rst -------------------------------------------------------------------------------- /examples/train_classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crowsonkb/torch-dist-utils/HEAD/examples/train_classifier.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crowsonkb/torch-dist-utils/HEAD/pyproject.toml -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crowsonkb/torch-dist-utils/HEAD/setup.cfg -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crowsonkb/torch-dist-utils/HEAD/setup.py -------------------------------------------------------------------------------- /torch_dist_utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crowsonkb/torch-dist-utils/HEAD/torch_dist_utils/__init__.py -------------------------------------------------------------------------------- /torch_dist_utils/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crowsonkb/torch-dist-utils/HEAD/torch_dist_utils/test.py -------------------------------------------------------------------------------- /torch_dist_utils/torch_dist_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crowsonkb/torch-dist-utils/HEAD/torch_dist_utils/torch_dist_utils.py -------------------------------------------------------------------------------- /torch_dist_utils/torchrun_slurm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/crowsonkb/torch-dist-utils/HEAD/torch_dist_utils/torchrun_slurm.py --------------------------------------------------------------------------------