You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
14 lines
344 B
14 lines
344 B
3 years ago
|
#!/bin/bash
|
||
|
|
|
||
|
|
config_path=$1
|
||
|
|
train_output_path=$2
|
||
|
|
|
||
|
|
FLAGS_cudnn_exhaustive_search=true \
|
||
|
|
FLAGS_conv_workspace_size_limit=4000 \
|
||
|
|
python ${BIN_DIR}/train.py \
|
||
|
|
--train-metadata=dump/train/norm/metadata.jsonl \
|
||
|
|
--dev-metadata=dump/dev/norm/metadata.jsonl \
|
||
|
|
--config=${config_path} \
|
||
|
|
--output-dir=${train_output_path} \
|
||
|
|
--nprocs=1
|