python.exe /tools/train.py -c ../configs/det/det_mv3_db.yml
Created by: Jasonxgw
2020-06-09 22:54:13,418-INFO: {'Global': {'algorithm': 'DB', 'use_gpu': True, 'epoch_num': 1200, 'log_smooth_window': 20, 'print_batch_step': 2, 'save_model_dir': '../output/det_db/', 'save_epoch_step': 200, 'eval_batch_step': 5000, 'train_batch_size_per_card': 16, 'test_batch_size_per_card': 16, 'image_shape': [3, 640, 640], 'reader_yml': '../configs/det/det_db_icdar15_reader.yml', 'pretrain_weights': '../pretrain_models/MobileNetV3_large_x0_5_pretrained/', 'checkpoints': None, 'save_res_path': '../output/det_db/predicts_db.txt', 'save_inference_dir': None}, 'Architecture': {'function': 'ppocr.modeling.architectures.det_model,DetModel'}, 'Backbone': {'function': 'ppocr.modeling.backbones.det_mobilenet_v3,MobileNetV3', 'scale': 0.5, 'model_name': 'large'}, 'Head': {'function': 'ppocr.modeling.heads.det_db_head,DBHead', 'model_name': 'large', 'k': 50, 'inner_channels': 96, 'out_channels': 2}, 'Loss': {'function': 'ppocr.modeling.losses.det_db_loss,DBLoss', 'balance_loss': True, 'main_loss_type': 'DiceLoss', 'alpha': 5, 'beta': 10, 'ohem_ratio': 3}, 'Optimizer': {'function': 'ppocr.optimizer,AdamDecay', 'base_lr': 0.001, 'beta1': 0.9, 'beta2': 0.999}, 'PostProcess': {'function': 'ppocr.postprocess.db_postprocess,DBPostProcess', 'thresh': 0.3, 'box_thresh': 0.7, 'max_candidates': 1000, 'unclip_ratio': 2.0}, 'TrainReader': {'reader_function': 'ppocr.data.det.dataset_traversal,TrainReader', 'process_function': 'ppocr.data.det.db_process,DBProcessTrain', 'num_workers': 8, 'img_set_dir': '../train_data/icdar2015/text_localization/', 'label_file_path': '../train_data/icdar2015/text_localization/train_icdar2015_label.txt'}, 'EvalReader': {'reader_function': 'ppocr.data.det.dataset_traversal,EvalTestReader', 'process_function': 'ppocr.data.det.db_process,DBProcessTest', 'img_set_dir': '../train_data/icdar2015/text_localization/', 'label_file_path': '../train_data/icdar2015/text_localization/test_icdar2015_label.txt', 'test_image_shape': [736, 1280]}, 'TestReader': {'reader_function': 'ppocr.data.det.dataset_traversal,EvalTestReader', 'process_function': 'ppocr.data.det.db_process,DBProcessTest', 'infer_img': None, 'img_set_dir': '../train_data/icdar2015/text_localization/', 'label_file_path': '../train_data/icdar2015/text_localization/test_icdar2015_label.txt', 'test_image_shape': [736, 1280], 'do_eval': True}} 3 640 640 3 640 640 2020-06-09 22:54:16,228-INFO: places would be ommited when DataLoader is not iterable 2020-06-09 22:54:18,197-INFO: Loading parameters from ../pretrain_models/MobileNetV3_large_x0_5_pretrained/... 2020-06-09 22:54:18,197-WARNING: ../pretrain_models/MobileNetV3_large_x0_5_pretrained/.pdparams not found, try to load model file saved with [ save_params, save_persistables, save_vars ] 2020-06-09 22:54:18,197-WARNING: ../pretrain_models/MobileNetV3_large_x0_5_pretrained/.pdparams not found, try to load model file saved with [ save_params, save_persistables, save_vars ] Traceback (most recent call last): File "C:\Users\Administrator\AppData\Local\Programs\Python\Python37\lib\site-packages\paddle\fluid\io.py", line 1865, in load_program_state filename=file_name) File "C:\Users\Administrator\AppData\Local\Programs\Python\Python37\lib\site-packages\paddle\fluid\io.py", line 793, in load_vars executor.run(load_prog) File "C:\Users\Administrator\AppData\Local\Programs\Python\Python37\lib\site-packages\paddle\fluid\executor.py", line 790, in run six.reraise(*sys.exc_info()) File "C:\Users\Administrator\AppData\Roaming\Python\Python37\site-packages\six.py", line 703, in reraise raise value File "C:\Users\Administrator\AppData\Local\Programs\Python\Python37\lib\site-packages\paddle\fluid\executor.py", line 785, in run use_program_cache=use_program_cache) File "C:\Users\Administrator\AppData\Local\Programs\Python\Python37\lib\site-packages\paddle\fluid\executor.py", line 838, in _run_impl use_program_cache=use_program_cache) File "C:\Users\Administrator\AppData\Local\Programs\Python\Python37\lib\site-packages\paddle\fluid\executor.py", line 912, in _run_program fetch_var_name) paddle.fluid.core_avx.EnforceNotMet:
C++ Call Stacks (More useful to developers):
Windows not support stack backtrace yet.
Python Call Stacks (More useful to users):
File "C:\Users\Administrator\AppData\Local\Programs\Python\Python37\lib\site-packages\paddle\fluid\framework.py", line 2525, in append_op attrs=kwargs.get("attrs", None)) File "C:\Users\Administrator\AppData\Local\Programs\Python\Python37\lib\site-packages\paddle\fluid\io.py", line 773, in load_vars attrs={'file_path': os.path.join(dirname, new_var.name)}) File "C:\Users\Administrator\AppData\Local\Programs\Python\Python37\lib\site-packages\paddle\fluid\io.py", line 1865, in load_program_state filename=file_name) File "D:\Downloads\PaddleOCR-develop\ppocr\utils\save_load.py", line 57, in _load_state state = fluid.io.load_program_state(path) File "D:\Downloads\PaddleOCR-develop\ppocr\utils\save_load.py", line 79, in load_params state = _load_state(path) File "D:\Downloads\PaddleOCR-develop\ppocr\utils\save_load.py", line 121, in init_model load_params(exe, program, path) File "D:/Downloads/PaddleOCR-develop/tools/train.py", line 76, in main init_model(config, train_program, exe) File "D:/Downloads/PaddleOCR-develop/tools/train.py", line 98, in main()
Error Message Summary:
InvalidArgumentError: tensor version 1974636042 is not supported, Only version 0 is supported [Hint: Expected version == 0U, but received version:1974636042 != 0U:0.] at (D:\1.7.2\paddle\paddle\fluid\framework\lod_tensor.cc:287) [operator < load > error]
During handling of the above exception, another exception occurred:
Traceback (most recent call last): File "D:/Downloads/PaddleOCR-develop/tools/train.py", line 98, in main() File "D:/Downloads/PaddleOCR-develop/tools/train.py", line 76, in main init_model(config, train_program, exe) File "D:\Downloads\PaddleOCR-develop\ppocr\utils\save_load.py", line 121, in init_model load_params(exe, program, path) File "D:\Downloads\PaddleOCR-develop\ppocr\utils\save_load.py", line 79, in load_params state = _load_state(path) File "D:\Downloads\PaddleOCR-develop\ppocr\utils\save_load.py", line 57, in _load_state state = fluid.io.load_program_state(path) File "C:\Users\Administrator\AppData\Local\Programs\Python\Python37\lib\site-packages\paddle\fluid\io.py", line 1868, in load_program_state "Failed to load model file , please make sure model file is saved with the " RuntimeError: Failed to load model file , please make sure model file is saved with the following APIs: save_params, save_persistables, save_vars W0609 22:54:16.984372 14860 device_context.cc:237] Please NOTE: device: 0, CUDA Capability: 61, Driver API Version: 10.1, Runtime API Version: 10.0 W0609 22:54:16.991372 14860 device_context.cc:245] device: 0, cuDNN Version: 7.6.