container_training/train.py
docker/train.py