yukkyo / Compare-PyTorch-Catalyst-Ignite-Lightning

Comparing PyTorch Catalyst, Ignite, Lightning by sample code

Comparing PyTorch Catalyst, Ignite, Lightning

Environment

• Python 3.7

torch==1.3.1
torchvision==0.4.2
catalyst==19.12
pytorch-ignite==0.2.1
pytorch-lightning==0.5.3.2

How to use

$ pip install -r requirements.txt

# Default
$ python train_default.py

# Catalyst
$ python train_catalyst.py

# Ignite
$ python train_ignite.py

# Lightning
$ python train_lightning.py