Skip to content

yukkyo/Compare-PyTorch-Catalyst-Ignite-Lightning

Repository files navigation

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

About

Comparing PyTorch Catalyst, Ignite, Lightning by sample code

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages