Skip to content

Instantly share code, notes, and snippets.

@mGalarnyk

mGalarnyk/CreateTrainerStartTraining.py

Created September 8, 2021 19:04
Show Gist options
  • Save mGalarnyk/4fe8cff7c2bbe292eecd8155e3ec0230 to your computer and use it in GitHub Desktop.
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