适配第三方模型参数的加载

adas-eye · Jun 12, 2020 · 7dd7ca4 · 7dd7ca4
1 parent 356662c
commit 7dd7ca4
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/config/rec_train_config.py b/config/rec_train_config.py
@@ -23,7 +23,8 @@
 ###
 rec_train_options = {
     # for train
-    'resume_from': None,
+    'resume_from': '/home/novio/project/PytorchOCR/rec_r34_vd_none_bilstm_ctc/best_accuracy.pdparams',
+    'third_party_name': 'paddle',
     'checkpoint_save_dir': "./out_dir/checkpoint",
     'device': 'cuda:0',
     'base_lr': 0.01,