From 2ad0ca44cd3bcc5896da1452185bf0600a15fd4e Mon Sep 17 00:00:00 2001
From: WenmuZhou <zjwenmu@gmail.com>
Date: Wed, 16 Dec 2020 00:00:03 +0800
Subject: [PATCH] update db config

---
 configs/det/det_mv3_db.yml    | 8 ++++----
 configs/det/det_r50_vd_db.yml | 2 +-
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/configs/det/det_mv3_db.yml b/configs/det/det_mv3_db.yml
index 36a6f755..a4ed569f 100644
--- a/configs/det/det_mv3_db.yml
+++ b/configs/det/det_mv3_db.yml
@@ -2,7 +2,7 @@ Global:
   use_gpu: true
   epoch_num: 1200
   log_smooth_window: 20
-  print_batch_step: 2
+  print_batch_step: 10
   save_model_dir: ./output/db_mv3/
   save_epoch_step: 1200
   # evaluation is run every 5000 iterations after the 4000th iteration
@@ -39,7 +39,7 @@ Loss:
   alpha: 5
   beta: 10
   ohem_ratio: 3
-  
+
 Optimizer:
   name: Adam
   beta1: 0.9
@@ -100,7 +100,7 @@ Train:
   loader:
     shuffle: True
     drop_last: False
-    batch_size_per_card: 4
+    batch_size_per_card: 16
     num_workers: 8
 
 Eval:
@@ -128,4 +128,4 @@ Eval:
     shuffle: False
     drop_last: False
     batch_size_per_card: 1 # must be 1
-    num_workers: 2
\ No newline at end of file
+    num_workers: 8
\ No newline at end of file
diff --git a/configs/det/det_r50_vd_db.yml b/configs/det/det_r50_vd_db.yml
index b70ab750..386a7970 100644
--- a/configs/det/det_r50_vd_db.yml
+++ b/configs/det/det_r50_vd_db.yml
@@ -6,7 +6,7 @@ Global:
   save_model_dir: ./output/det_r50_vd/
   save_epoch_step: 1200
   # evaluation is run every 5000 iterations after the 4000th iteration
-  eval_batch_step: [5000,4000]
+  eval_batch_step: [4000,5000]
   # if pretrained_model is saved in static mode, load_static_weights must set to True
   load_static_weights: True
   cal_metric_during_train: False
-- 
GitLab