Update Dockerfile for GPU now that the branch has been merged (tensor…

…flow#3576) * Remove cd and checkout * Updating comment
vijethrai · Mar 13, 2018 · 4c7c8fa · 4c7c8fa
1 parent 69c7559
commit 4c7c8fa
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/official/Dockerfile.cpu b/official/Dockerfile.cpu
@@ -12,6 +12,6 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
 
 RUN pip install tf-nightly
 
-# Checkout tensorflow/models and pin to a particular version
+# Checkout tensorflow/models at HEAD
 RUN git clone https://github.com/tensorflow/models.git /tensorflow_models
 
diff --git a/official/Dockerfile.gpu b/official/Dockerfile.gpu
@@ -12,7 +12,7 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
 
 RUN pip install tf-nightly-gpu
 
-# Checkout tensorflow/models and pin to a particular version
-RUN git clone https://github.com/tensorflow/models.git /tensorflow_models && \
-    cd /tensorflow_models && git checkout feat/multi-gpu-resnet
+# Checkout tensorflow/models at HEAD
+RUN git clone https://github.com/tensorflow/models.git /tensorflow_models
+