add docker setup for training with tensorboard status

models/.dockerignore

@@ -0,0 +1,4 @@
+*.env
+data
+logs
+configs

models/Dockerfile

@@ -0,0 +1,11 @@
+FROM python:3.7
+
+WORKDIR /usr/src/app
+
+RUN apt-get update 
+RUN apt-get install libgl1 -y
+
+COPY requirements.txt ./
+RUN pip install --no-cache-dir -r requirements.txt
+
+COPY . .

models/docker-compose.yml

@@ -0,0 +1,16 @@
+version: "3"
+services:
+ model:
+ build: .
+ volumes:
+ - ./logs/:/usr/src/app/logs:z
+ - ./data/:/usr/src/app/data:z
+ - ./configs:/usr/src/app/configs:z
+ command: sh -c "python train.py logger=tensorboard"
+ tensorboard:
+ image: tensorflow/tensorflow
+ volumes:
+ - ./logs:/logs:z
+ command: sh -c "tensorboard --logdir /logs --bind_all"
+ ports:
+ - 6006:6006