/ray-pytorch-lightning

Example MNIST training using the following tech stack: Ray, PyTorch Lightning, PyTorch and Python 3.10.12.

Primary LanguagePythonMIT LicenseMIT

Watchers