add dockerfile for model

model/.dockerignore

@@ -0,0 +1,4 @@
+.env
+in/
+out/
+new_proect.py

model/Dockerfile

@@ -0,0 +1,13 @@
+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 . .
+
+CMD sh -c "python train.py --config config.json"

model/docker-compose.yml

@@ -0,0 +1,12 @@
+version: "3" # optional since v1.27.0
+services:
+ model:
+ build: .
+ ports:
+ - 6006:6006
+ volumes:
+ - ./saved:/usr/src/app/saved:z
+ # - ./out/:/usr/src/app/out:z
+ # - ./in/:/usr/src/app/in:z
+ #env_file:
+ # - .env

model/requirements.txt

@@ -2,4 +2,5 @@ torch>=1.1
 torchvision
 numpy
 tqdm
-tensorboard>=1.14
+tensorboardx
+pandas