This directory contains a dockerfile to build the training container, a python script to include in the container, and a YAML file to provide user input to the training process.
The dockerfile is built on tensorflow/tensorflow:latest-gpu-jupyter. It adds pandas, ktrain, and h5py.
docker build -t train -f training.dockerfile .
nvidia-docker run -dit --name train -v /root/sets:/data/sets -v/root/models:/data/models tensorflow/tensorflow:latest-gpu-jupyter python3 /training.py