Commit 40b7aaf5 authored by bitsoko's avatar bitsoko Committed by GitHub

Update runCpu.sh

parent 6cf11eee
......@@ -27,7 +27,7 @@ fi
#prepare data
#python prepro.py --train_image_dir="${IMG_TRAIN_DIR}" --val_image_dir="${IMG_VAL_DIR}" --train_captions_file="${CAPTIONS_DIR}/val.json" --val_captions_file="${CAPTIONS_DIR}/val.json" --output_dir="${OUTPUT_DIR}/tf" --word_counts_output_file="${OUTPUT_DIR}/word_counts.txt"
python prepro.py --train_image_dir="${IMG_TRAIN_DIR}" --val_image_dir="${IMG_VAL_DIR}" --train_captions_file="${CAPTIONS_DIR}/val.json" --val_captions_file="${CAPTIONS_DIR}/val.json" --output_dir="${OUTPUT_DIR}/tf" --word_counts_output_file="${OUTPUT_DIR}/word_counts.txt"
python train.py \
--input_file_pattern="${MSCOCO_DIR}/tf/val-?????-of-00256" \
......
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