mGalarnyk/CreateTrainerStartTraining.py

Created September 8, 2021 19:04

Star () You must be signed in to star a gist
Fork () You must be signed in to fork a gist

Learn more about clone URLs
Clone this repository at <script src="https://gist.github.com/mGalarnyk/4fe8cff7c2bbe292eecd8155e3ec0230.js"></script>
Save mGalarnyk/4fe8cff7c2bbe292eecd8155e3ec0230 to your computer and use it in GitHub Desktop.

Download ZIP

Gist originally from Introducing Ray Lightning: Multi-node PyTorch Lightning Training made easy: https://www.anyscale.com/blog/introducing-ray-lightning-multi-node-gpu-training-for-pytorch-lightning-made

Raw

CreateTrainerStartTraining.py

	model = LightningMNISTClassifier(config, data_dir="./")

	trainer = pl.Trainer( max_epochs=10)
	trainer.fit(model)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment