/pytorch_lightning_playground

Primary LanguagePythonMIT LicenseMIT

PyTorch lightning playground

Usage

# install dependencies
$ pip install -r requirements.txt

# train
$ python train_mnist.py

# view log with tensorboard
$ tensorboard --logdir logs/