Commit 7e9b9830 authored by Chris Shallue's avatar Chris Shallue Committed by GitHub
Browse files

Merge pull request #481 from cshallue/readme

Remove flag --config=cuda. It's not necessary and can cause a warning.
parents 19d80b98 d7a4970e
......@@ -218,7 +218,7 @@ INCEPTION_CHECKPOINT="${HOME}/im2txt/data/inception_v3.ckpt"
MODEL_DIR="${HOME}/im2txt/model"
# Build the model.
bazel build -c opt --config=cuda im2txt/...
bazel build -c opt im2txt/...
# Run the training script.
bazel-bin/im2txt/train \
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment