Skip to content

GitLab

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
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
im2txt/README.md
View file @ d7a4970e
...@@ -218,7 +218,7 @@ INCEPTION_CHECKPOINT="${HOME}/im2txt/data/inception_v3.ckpt" ...@@ -218,7 +218,7 @@ INCEPTION_CHECKPOINT="${HOME}/im2txt/data/inception_v3.ckpt"
MODEL_DIR="${HOME}/im2txt/model" MODEL_DIR="${HOME}/im2txt/model"
# Build the model. # Build the model.
bazel build -c opt --config=cuda im2txt/... bazel build -c opt im2txt/...
# Run the training script. # Run the training script.
bazel-bin/im2txt/train \ 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