From 4c5487f28caa503f1dbbcd2a4e0845ccda50e151 Mon Sep 17 00:00:00 2001 From: Kaipeng Deng Date: Wed, 18 Mar 2020 12:16:06 +0800 Subject: [PATCH] add YOLOv3Loss.batch_size comments (#348) --- configs/dcn/yolov3_r50vd_dcn.yml | 4 ++++ .../yolov3_r50vd_dcn_db_iouloss_obj365_pretrained_coco.yml | 4 ++++ configs/dcn/yolov3_r50vd_dcn_db_obj365_pretrained_coco.yml | 4 ++++ configs/dcn/yolov3_r50vd_dcn_obj365_pretrained_coco.yml | 4 ++++ configs/yolov3_darknet.yml | 4 ++++ configs/yolov3_darknet_voc.yml | 4 ++++ configs/yolov3_mobilenet_v1.yml | 4 ++++ configs/yolov3_mobilenet_v1_fruit.yml | 4 ++++ configs/yolov3_mobilenet_v1_voc.yml | 4 ++++ configs/yolov3_r34.yml | 4 ++++ configs/yolov3_r34_voc.yml | 4 ++++ 11 files changed, 44 insertions(+) diff --git a/configs/dcn/yolov3_r50vd_dcn.yml b/configs/dcn/yolov3_r50vd_dcn.yml index e16aac604..0493597b1 100755 --- a/configs/dcn/yolov3_r50vd_dcn.yml +++ b/configs/dcn/yolov3_r50vd_dcn.yml @@ -40,6 +40,10 @@ YOLOv3Head: score_threshold: 0.01 YOLOv3Loss: + # batch_size here is only used for fine grained loss, not used + # for training batch_size setting, training batch_size setting + # is in configs/yolov3_reader.yml TrainReader.batch_size, batch + # size here should be set as same value as TrainReader.batch_size batch_size: 8 ignore_thresh: 0.7 label_smooth: false diff --git a/configs/dcn/yolov3_r50vd_dcn_db_iouloss_obj365_pretrained_coco.yml b/configs/dcn/yolov3_r50vd_dcn_db_iouloss_obj365_pretrained_coco.yml index 8a7fde797..5e9431453 100755 --- a/configs/dcn/yolov3_r50vd_dcn_db_iouloss_obj365_pretrained_coco.yml +++ b/configs/dcn/yolov3_r50vd_dcn_db_iouloss_obj365_pretrained_coco.yml @@ -42,6 +42,10 @@ YOLOv3Head: drop_block: true YOLOv3Loss: + # batch_size here is only used for fine grained loss, not used + # for training batch_size setting, training batch_size setting + # is in configs/yolov3_reader.yml TrainReader.batch_size, batch + # size here should be set as same value as TrainReader.batch_size batch_size: 8 ignore_thresh: 0.7 label_smooth: false diff --git a/configs/dcn/yolov3_r50vd_dcn_db_obj365_pretrained_coco.yml b/configs/dcn/yolov3_r50vd_dcn_db_obj365_pretrained_coco.yml index 615eb0c22..3c69e410a 100755 --- a/configs/dcn/yolov3_r50vd_dcn_db_obj365_pretrained_coco.yml +++ b/configs/dcn/yolov3_r50vd_dcn_db_obj365_pretrained_coco.yml @@ -43,6 +43,10 @@ YOLOv3Head: keep_prob: 0.94 YOLOv3Loss: + # batch_size here is only used for fine grained loss, not used + # for training batch_size setting, training batch_size setting + # is in configs/yolov3_reader.yml TrainReader.batch_size, batch + # size here should be set as same value as TrainReader.batch_size batch_size: 8 ignore_thresh: 0.7 label_smooth: false diff --git a/configs/dcn/yolov3_r50vd_dcn_obj365_pretrained_coco.yml b/configs/dcn/yolov3_r50vd_dcn_obj365_pretrained_coco.yml index 7042d9538..014a7947e 100755 --- a/configs/dcn/yolov3_r50vd_dcn_obj365_pretrained_coco.yml +++ b/configs/dcn/yolov3_r50vd_dcn_obj365_pretrained_coco.yml @@ -41,6 +41,10 @@ YOLOv3Head: score_threshold: 0.01 YOLOv3Loss: + # batch_size here is only used for fine grained loss, not used + # for training batch_size setting, training batch_size setting + # is in configs/yolov3_reader.yml TrainReader.batch_size, batch + # size here should be set as same value as TrainReader.batch_size batch_size: 8 ignore_thresh: 0.7 label_smooth: false diff --git a/configs/yolov3_darknet.yml b/configs/yolov3_darknet.yml index 2ea5ba7bd..9e91e7eb2 100644 --- a/configs/yolov3_darknet.yml +++ b/configs/yolov3_darknet.yml @@ -35,6 +35,10 @@ YOLOv3Head: score_threshold: 0.01 YOLOv3Loss: + # batch_size here is only used for fine grained loss, not used + # for training batch_size setting, training batch_size setting + # is in configs/yolov3_reader.yml TrainReader.batch_size, batch + # size here should be set as same value as TrainReader.batch_size batch_size: 8 ignore_thresh: 0.7 label_smooth: true diff --git a/configs/yolov3_darknet_voc.yml b/configs/yolov3_darknet_voc.yml index 241d12934..b1c48f5f6 100644 --- a/configs/yolov3_darknet_voc.yml +++ b/configs/yolov3_darknet_voc.yml @@ -36,6 +36,10 @@ YOLOv3Head: score_threshold: 0.01 YOLOv3Loss: + # batch_size here is only used for fine grained loss, not used + # for training batch_size setting, training batch_size setting + # is in configs/yolov3_reader.yml TrainReader.batch_size, batch + # size here should be set as same value as TrainReader.batch_size batch_size: 8 ignore_thresh: 0.7 label_smooth: false diff --git a/configs/yolov3_mobilenet_v1.yml b/configs/yolov3_mobilenet_v1.yml index a66b26c72..b3e3eebe8 100644 --- a/configs/yolov3_mobilenet_v1.yml +++ b/configs/yolov3_mobilenet_v1.yml @@ -36,6 +36,10 @@ YOLOv3Head: score_threshold: 0.01 YOLOv3Loss: + # batch_size here is only used for fine grained loss, not used + # for training batch_size setting, training batch_size setting + # is in configs/yolov3_reader.yml TrainReader.batch_size, batch + # size here should be set as same value as TrainReader.batch_size batch_size: 8 ignore_thresh: 0.7 label_smooth: true diff --git a/configs/yolov3_mobilenet_v1_fruit.yml b/configs/yolov3_mobilenet_v1_fruit.yml index 74b1bb703..78f50206e 100644 --- a/configs/yolov3_mobilenet_v1_fruit.yml +++ b/configs/yolov3_mobilenet_v1_fruit.yml @@ -38,6 +38,10 @@ YOLOv3Head: score_threshold: 0.01 YOLOv3Loss: + # batch_size here is only used for fine grained loss, not used + # for training batch_size setting, training batch_size setting + # is in configs/yolov3_reader.yml TrainReader.batch_size, batch + # size here should be set as same value as TrainReader.batch_size batch_size: 8 ignore_thresh: 0.7 label_smooth: true diff --git a/configs/yolov3_mobilenet_v1_voc.yml b/configs/yolov3_mobilenet_v1_voc.yml index 2ce5e40bf..1b7097ad3 100644 --- a/configs/yolov3_mobilenet_v1_voc.yml +++ b/configs/yolov3_mobilenet_v1_voc.yml @@ -37,6 +37,10 @@ YOLOv3Head: score_threshold: 0.01 YOLOv3Loss: + # batch_size here is only used for fine grained loss, not used + # for training batch_size setting, training batch_size setting + # is in configs/yolov3_reader.yml TrainReader.batch_size, batch + # size here should be set as same value as TrainReader.batch_size batch_size: 8 ignore_thresh: 0.7 label_smooth: false diff --git a/configs/yolov3_r34.yml b/configs/yolov3_r34.yml index 4cc05f24d..db2436643 100644 --- a/configs/yolov3_r34.yml +++ b/configs/yolov3_r34.yml @@ -38,6 +38,10 @@ YOLOv3Head: score_threshold: 0.01 YOLOv3Loss: + # batch_size here is only used for fine grained loss, not used + # for training batch_size setting, training batch_size setting + # is in configs/yolov3_reader.yml TrainReader.batch_size, batch + # size here should be set as same value as TrainReader.batch_size batch_size: 8 ignore_thresh: 0.7 label_smooth: true diff --git a/configs/yolov3_r34_voc.yml b/configs/yolov3_r34_voc.yml index f1cb5a4a2..2d980dd0c 100644 --- a/configs/yolov3_r34_voc.yml +++ b/configs/yolov3_r34_voc.yml @@ -39,6 +39,10 @@ YOLOv3Head: score_threshold: 0.01 YOLOv3Loss: + # batch_size here is only used for fine grained loss, not used + # for training batch_size setting, training batch_size setting + # is in configs/yolov3_reader.yml TrainReader.batch_size, batch + # size here should be set as same value as TrainReader.batch_size batch_size: 8 ignore_thresh: 0.7 label_smooth: false -- GitLab