add pretrain_models

4 years ago · 02da7ec53e
parent c654dbf709
commit 02da7ec53e
1 changed files with 1 additions and 1 deletions
--- a/configs/det/det_r50_vd_db.yml
+++ b/configs/det/det_r50_vd_db.yml
@ -10,7 +10,7 @@ Global:
  # if pretrained_model is saved in static mode, load_static_weights must set to True
  load_static_weights: True
  cal_metric_during_train: False
-  pretrained_model:
+  pretrained_model: ./pretrain_models/ResNet50_vd_ssld_pretrained
  checkpoints:
  save_inference_dir:
  use_visualdl: False