diff --git a/lib/models/include/models/yolov10/yolov10.h b/lib/models/include/models/yolov10/yolov10.h index 567af5eeba..da46794414 100644 --- a/lib/models/include/models/yolov10/yolov10.h +++ b/lib/models/include/models/yolov10/yolov10.h @@ -143,10 +143,10 @@ tensor_guid_t create_yolov10_c2fcib_module( /** * \brief Creates layers matching a single iteration of - * + * * the loop * - * that creates the box head (aka cv3) for + * that creates the box head (aka cv2) for * * ultralytics' v10Detect module. * @@ -159,10 +159,10 @@ tensor_guid_t /** * \brief Creates layers matching a single iteration of - * + * * the loop * - * that creates the class head (aka cls_head, or cv2) for + * that creates the class head (aka cls_head, or cv3) for * * ultralytics' v10Detect module. * diff --git a/lib/models/src/models/yolov10/yolov10.cc b/lib/models/src/models/yolov10/yolov10.cc index d980c21098..b1e2950c7d 100644 --- a/lib/models/src/models/yolov10/yolov10.cc +++ b/lib/models/src/models/yolov10/yolov10.cc @@ -355,13 +355,16 @@ tensor_guid_t /** * https://github.com/ultralytics/ultralytics/blob/f8ad132a15b5f6818c2ce0647b40dc57e993bf0c/ultralytics/nn/modules/conv.py#L66 + * + * Note that eps and momentum are set in: + * https://github.com/ultralytics/ultralytics/blob/f8ad132a15b5f6818c2ce0647b40dc57e993bf0c/ultralytics/utils/torch_utils.py#L558-L568 */ tensor_guid_t out = cgb.batch_norm( /*input=*/conv, /*affine=*/true, /*activation=*/std::nullopt, - /*eps=*/1e-5, - /*momentum=*/0.1); + /*eps=*/1e-3, + /*momentum=*/0.03); /** * https://github.com/ultralytics/ultralytics/blob/f8ad132a15b5f6818c2ce0647b40dc57e993bf0c/ultralytics/nn/modules/conv.py#L67