From 56cc45a95eadffbc102b4c9db00c08cb182a08ad Mon Sep 17 00:00:00 2001
From: Arne Binder <arne.b.binder@gmail.com>
Date: Thu, 26 Apr 2018 16:01:24 +0200
Subject: [PATCH] fixed docker-compose usage for training

---
 docker/tools/visualize/Dockerfile.tf1_3_gpu | 1 +
 1 file changed, 1 insertion(+)

diff --git a/docker/tools/visualize/Dockerfile.tf1_3_gpu b/docker/tools/visualize/Dockerfile.tf1_3_gpu
index 546b8ec..0778c8f 100644
--- a/docker/tools/visualize/Dockerfile.tf1_3_gpu
+++ b/docker/tools/visualize/Dockerfile.tf1_3_gpu
@@ -172,4 +172,5 @@ VOLUME /root/corpora_out
 #ENTRYPOINT ["/root/recursive-embedding/set-user-with-folder.sh", "/root/train", "python", "train_fold.py"]
 
 # pass as env variable
+ARG NVIDIA_VISIBLE_DEVICES=0
 ENV NVIDIA_VISIBLE_DEVICES="$NVIDIA_VISIBLE_DEVICES"