- 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
$ 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