From 6de43fbb471b91e185512d2e60c0c5d0ddd78a25 Mon Sep 17 00:00:00 2001 From: tink2123 Date: Tue, 19 May 2020 17:30:22 +0800 Subject: [PATCH] update check_gpu --- configs/rec/rec_chinese_lite_train.yml | 2 +- tools/eval.py | 2 +- tools/train.py | 4 ++-- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/configs/rec/rec_chinese_lite_train.yml b/configs/rec/rec_chinese_lite_train.yml index 76f7f2c0..ec1b7a69 100755 --- a/configs/rec/rec_chinese_lite_train.yml +++ b/configs/rec/rec_chinese_lite_train.yml @@ -15,7 +15,7 @@ Global: character_dict_path: ./ppocr/utils/ppocr_keys_v1.txt loss_type: ctc reader_yml: ./configs/rec/rec_chinese_reader.yml - pretrain_weights: ./pretrain_models/CRNN/best_accuracy + pretrain_weights: checkpoints: save_inference_dir: Architecture: diff --git a/tools/eval.py b/tools/eval.py index 304a1340..fcb572c6 100755 --- a/tools/eval.py +++ b/tools/eval.py @@ -51,7 +51,7 @@ def main(): # check if set use_gpu=True in paddlepaddle cpu version use_gpu = config['Global']['use_gpu'] - program.check_gpu(True) + program.check_gpu(use_gpu) alg = config['Global']['algorithm'] assert alg in ['EAST', 'DB', 'Rosetta', 'CRNN', 'STARNet', 'RARE'] diff --git a/tools/train.py b/tools/train.py index 08c5a641..4f69d40f 100755 --- a/tools/train.py +++ b/tools/train.py @@ -31,7 +31,7 @@ def set_paddle_flags(**kwargs): # NOTE(paddle-dev): All of these flags should be # set before `import paddle`. Otherwise, it would -# not take any effect. +# not take any effect. set_paddle_flags( FLAGS_eager_delete_tensor_gb=0, # enable GC to save memory ) @@ -52,7 +52,7 @@ def main(): # check if set use_gpu=True in paddlepaddle cpu version use_gpu = config['Global']['use_gpu'] - program.check_gpu(True) + program.check_gpu(use_gpu) alg = config['Global']['algorithm'] assert alg in ['EAST', 'DB', 'Rosetta', 'CRNN', 'STARNet', 'RARE'] -- GitLab