motokimura / pytorch_lightning_playground

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

PyTorch lightning playground

Usage

# install dependencies
$ pip install -r requirements.txt

# train
$ python train_mnist.py
# view log with tensorboard
$ tensorboard --logdir logs/

About

License:MIT License


Languages

Language:Python 100.0%