Commit d7a4970e authored by Christopher Shallue's avatar Christopher Shallue
Browse files

Remove flag --config=cuda. It's not necessary and can cause a warning.

parent 19d80b98
......@@ -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