Skip to content

Latest commit

 

History

History
33 lines (22 loc) · 407 Bytes

File metadata and controls

33 lines (22 loc) · 407 Bytes

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