diff --git a/configs/det/det_mv3_db.yml b/configs/det/det_mv3_db.yml index 36a6f755383e525a8a496b060465cf027f3f31f8..a4ed569f4aeae55977693e033fd5b12fa741a1e4 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 b70ab7505a4c1970f084b6c02233526d7f7188b9..386a79705dd0f059c5ee366ab8bb914eb59ddb32 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