pytorch-lightning-tutorial

Pytorch lightning tutorial using MNIST

Youtube stream (Maybe there will be another… still planning!)

Pytorch lightning introduction

scheduler introduction (Japanese)

Installation

Python>=3.7, creation using anaconda is recommended. Install libraries by pip install -r requirements.txt.

Train MNIST

Run (example)

python train.py --root_dir "./"

It will download the dataset to root_dir and start training. You can monitor the training process by launching tensorboard in another terminal:

tensorboard --logdir logs

And go to localhost:6006 in your browser.

GitHub

View Github