diff --git a/configs/rec/rec_chinese_lite_train.yml b/configs/rec/rec_chinese_lite_train.yml index 29506ff4e48d5caa4c4100d3186d7b03f778f0ab..e821a6230b9b2a66b57a974f28beb4e19ba61aa3 100755 --- a/configs/rec/rec_chinese_lite_train.yml +++ b/configs/rec/rec_chinese_lite_train.yml @@ -1,6 +1,5 @@ Global: algorithm: CRNN - dataset: common use_gpu: true epoch_num: 300 log_smooth_window: 20 diff --git a/set_env.sh b/set_env.sh new file mode 100644 index 0000000000000000000000000000000000000000..b32e58e3d2ab8ac99759d1acc16b0d3598fa26f9 --- /dev/null +++ b/set_env.sh @@ -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↩ diff --git a/tools/eval.py b/tools/eval.py index 14840f726873f75964596b9548b0bf068bd967af..3b176648812fff74e9878e0b789b6d6f66fc70f5 100755 --- a/tools/eval.py +++ b/tools/eval.py @@ -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, \