Skip to content

Commit

Permalink
Update example.sh
Browse files Browse the repository at this point in the history
  • Loading branch information
jiahuei committed Jun 5, 2019
1 parent fc4194a commit bf0fe92
Showing 1 changed file with 13 additions and 5 deletions.
18 changes: 13 additions & 5 deletions src/example.sh
Original file line number Diff line number Diff line change
Expand Up @@ -4,24 +4,32 @@ DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" >/dev/null 2>&1 && pwd )"
cd ${DIR}

### Training
# Default dataset directory
# Default
python train.py

# Custom dataset directory
# Custom MS-COCO directory
python train.py \
--dataset_dir '/home/jiahuei/Documents/3_Datasets/MSCOCO_captions'

# Word token, custom dataset directory, GPU 1
# Word token, custom MS-COCO directory, GPU 1
python train.py \
--token_type 'word' \
--dataset_dir '/home/jiahuei/Documents/3_Datasets/MSCOCO_captions' \
--gpu '1'

# InstaPIC
python train.py \
--dataset_file_pattern 'insta_{}_v25595_s15'

# Custom InstaPIC directory
python train.py \
--dataset_file_pattern 'insta_{}_v25595_s15' \
--dataset_dir '/home/jiahuei/Documents/3_Datasets/InstaPIC'


### Inference
# Default dataset and checkpoint directories
python infer.py \
--infer_checkpoints_dir 'mscoco/radix_add_softmax_h8_tie_lstm_run_01'
python infer.py

# Custom dataset and checkpoint directories
python infer.py \
Expand Down

0 comments on commit bf0fe92

Please sign in to comment.