add pretrain_models

02da7ec5 · WenmuZhou · c654dbf7 · 02da7ec5
Commit 02da7ec5 authored Dec 09, 2020 by WenmuZhou
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

configs/det/det_r50_vd_db.yml configs/det/det_r50_vd_db.yml +1 -1

No files found.
--- 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