diff --git a/docker/tools/visualize/Dockerfile.tf1_3_gpu b/docker/tools/visualize/Dockerfile.tf1_3_gpu
index 16057ae..07b424e 100644
--- a/docker/tools/visualize/Dockerfile.tf1_3_gpu
+++ b/docker/tools/visualize/Dockerfile.tf1_3_gpu
@@ -166,7 +166,8 @@ COPY src "$PROJECT_ROOT"/src
 
 WORKDIR "$PROJECT_ROOT"/src
 
-#VOLUME /root/train
+VOLUME /root/train
+VOLUME /root/corpora_out
 
 #ENTRYPOINT ["/root/recursive-embedding/set-user-with-folder.sh", "/root/train", "python", "train_fold.py"]