├── .gitignore ├── README.md ├── environment ├── Dockerfile ├── environment.yml ├── install_conda_env.sh ├── install_docker_env.sh └── keras.json ├── example_configs ├── cifar10.yml ├── cifar100.yml └── mnist.yml ├── mutli-gpu-keras-tf ├── __init__.py ├── data_iterators.py ├── model.py ├── networks.py ├── setup.py └── trainer.py ├── readme_imgs ├── cifar100_train_loss.png ├── cifar10_train_loss.png └── mnist_train_loss.png ├── scripts ├── evaluate.py └── train.py └── tests ├── test_data_iterators.py └── test_networks.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sallamander/multi-gpu-keras-tf/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sallamander/multi-gpu-keras-tf/HEAD/README.md -------------------------------------------------------------------------------- /environment/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sallamander/multi-gpu-keras-tf/HEAD/environment/Dockerfile -------------------------------------------------------------------------------- /environment/environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sallamander/multi-gpu-keras-tf/HEAD/environment/environment.yml -------------------------------------------------------------------------------- /environment/install_conda_env.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sallamander/multi-gpu-keras-tf/HEAD/environment/install_conda_env.sh -------------------------------------------------------------------------------- /environment/install_docker_env.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sallamander/multi-gpu-keras-tf/HEAD/environment/install_docker_env.sh -------------------------------------------------------------------------------- /environment/keras.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sallamander/multi-gpu-keras-tf/HEAD/environment/keras.json -------------------------------------------------------------------------------- /example_configs/cifar10.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sallamander/multi-gpu-keras-tf/HEAD/example_configs/cifar10.yml -------------------------------------------------------------------------------- /example_configs/cifar100.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sallamander/multi-gpu-keras-tf/HEAD/example_configs/cifar100.yml -------------------------------------------------------------------------------- /example_configs/mnist.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sallamander/multi-gpu-keras-tf/HEAD/example_configs/mnist.yml -------------------------------------------------------------------------------- /mutli-gpu-keras-tf/__init__.py: -------------------------------------------------------------------------------- 1 | """dist-keras-tf package""" 2 | -------------------------------------------------------------------------------- /mutli-gpu-keras-tf/data_iterators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sallamander/multi-gpu-keras-tf/HEAD/mutli-gpu-keras-tf/data_iterators.py -------------------------------------------------------------------------------- /mutli-gpu-keras-tf/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sallamander/multi-gpu-keras-tf/HEAD/mutli-gpu-keras-tf/model.py -------------------------------------------------------------------------------- /mutli-gpu-keras-tf/networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sallamander/multi-gpu-keras-tf/HEAD/mutli-gpu-keras-tf/networks.py -------------------------------------------------------------------------------- /mutli-gpu-keras-tf/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sallamander/multi-gpu-keras-tf/HEAD/mutli-gpu-keras-tf/setup.py -------------------------------------------------------------------------------- /mutli-gpu-keras-tf/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sallamander/multi-gpu-keras-tf/HEAD/mutli-gpu-keras-tf/trainer.py -------------------------------------------------------------------------------- /readme_imgs/cifar100_train_loss.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sallamander/multi-gpu-keras-tf/HEAD/readme_imgs/cifar100_train_loss.png -------------------------------------------------------------------------------- /readme_imgs/cifar10_train_loss.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sallamander/multi-gpu-keras-tf/HEAD/readme_imgs/cifar10_train_loss.png -------------------------------------------------------------------------------- /readme_imgs/mnist_train_loss.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sallamander/multi-gpu-keras-tf/HEAD/readme_imgs/mnist_train_loss.png -------------------------------------------------------------------------------- /scripts/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sallamander/multi-gpu-keras-tf/HEAD/scripts/evaluate.py -------------------------------------------------------------------------------- /scripts/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sallamander/multi-gpu-keras-tf/HEAD/scripts/train.py -------------------------------------------------------------------------------- /tests/test_data_iterators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sallamander/multi-gpu-keras-tf/HEAD/tests/test_data_iterators.py -------------------------------------------------------------------------------- /tests/test_networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sallamander/multi-gpu-keras-tf/HEAD/tests/test_networks.py --------------------------------------------------------------------------------