mv dataset in config
This commit is contained in:
parent
56c6c3ae0e
commit
476c51a73c
|
@ -1,6 +1,5 @@
|
|||
Global:
|
||||
algorithm: CRNN
|
||||
dataset: common
|
||||
use_gpu: true
|
||||
epoch_num: 300
|
||||
log_smooth_window: 20
|
||||
|
|
|
@ -0,0 +1,12 @@
|
|||
#. /paddle/set_env.sh↩
|
||||
export CUDA_VISIBLE_DEVICES="0,1,2,3"↩
|
||||
export PYTHONPATH=$PYTHONPATH:.↩
|
||||
export FLAGS_fraction_of_gpu_memory_to_use=1.0↩
|
||||
↩
|
||||
python_bin_dir="/opt/_internal/cpython-3.7.0/bin/"↩
|
||||
alias python=$python_bin_dir"python3.7"↩
|
||||
alias pip=$python_bin_dir"pip3.7"↩
|
||||
alias ipython=$python_bin_dir"ipython3"↩
|
||||
export LD_LIBRARY_PATH=/opt/_internal/cpython-3.7.0/lib:$LD_LIBRARY_PATH↩
|
||||
export PYTHONPATH=$PYTHONPATH:.↩
|
||||
ldconfig↩
|
|
@ -79,9 +79,8 @@ def main():
|
|||
'fetch_varname_list':eval_fetch_varname_list}
|
||||
metrics = eval_det_run(exe, config, eval_info_dict, "test")
|
||||
else:
|
||||
dataset = config['Global']['dataset']
|
||||
assert dataset in ['lmdb', 'common']
|
||||
if dataset == 'common':
|
||||
reader_type = config['Global']['reader_yml']
|
||||
if "chinese" in reader_type:
|
||||
eval_reader = reader_main(config=config, mode="eval")
|
||||
eval_info_dict = {'program': eval_program, \
|
||||
'reader': eval_reader, \
|
||||
|
|
Loading…
Reference in New Issue