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.
15 lines
363 B
15 lines
363 B
#!/bin/bash
|
|
|
|
config_path=$1
|
|
train_output_path=$2
|
|
ckpt_name=$3
|
|
test_input=$4
|
|
|
|
FLAGS_allocator_strategy=naive_best_fit \
|
|
FLAGS_fraction_of_gpu_memory_to_use=0.01 \
|
|
python3 ${BIN_DIR}/synthesize.py \
|
|
--config=${config_path} \
|
|
--checkpoint=${train_output_path}/checkpoints/${ckpt_name} \
|
|
--input=${test_input} \
|
|
--output-dir=${train_output_path}/test
|