LICENSE
README.md
setup.py
xpi_nn_train/__init__.py
xpi_nn_train/utils.py
xpi_nn_train.egg-info/PKG-INFO
xpi_nn_train.egg-info/SOURCES.txt
xpi_nn_train.egg-info/dependency_links.txt
xpi_nn_train.egg-info/requires.txt
xpi_nn_train.egg-info/top_level.txt
xpi_nn_train/datasets/lightning.py
xpi_nn_train/datasets/torchvision.py
xpi_nn_train/examples/finetune_cifar10_mbv3.py
xpi_nn_train/examples/finetune_cifar10_mbv3_ddp.py
xpi_nn_train/examples/train_mnist.py
xpi_nn_train/models/espcn.py
xpi_nn_train/models/lenet.py
xpi_nn_train/models/mlp.py
xpi_nn_train/models/torchvision.py
xpi_nn_train/trainers/__init__.py
xpi_nn_train/trainers/lightning.py
xpi_nn_train/trainers/torch.py
xpi_nn_train/trainers/torch_ddp.py