From 66eb8680932475ad034c03862276483cb2f8b464 Mon Sep 17 00:00:00 2001 From: kprokofi Date: Wed, 13 Nov 2024 21:26:49 +0900 Subject: [PATCH 1/2] added default template --- .../templates/keypoint_detection/rtmpose_tiny/template.yaml | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/src/otx/tools/templates/keypoint_detection/rtmpose_tiny/template.yaml b/src/otx/tools/templates/keypoint_detection/rtmpose_tiny/template.yaml index 08e5567b9a4..81c2efef1c3 100644 --- a/src/otx/tools/templates/keypoint_detection/rtmpose_tiny/template.yaml +++ b/src/otx/tools/templates/keypoint_detection/rtmpose_tiny/template.yaml @@ -23,3 +23,7 @@ training_targets: # Computational Complexity gigaflops: 1.3 size: 3.1 + +# Model spec +model_category: BALANCE +is_default_for_task: true From 7008c699fb0de23db3cfdf9241be136ab9005d1a Mon Sep 17 00:00:00 2001 From: kprokofi Date: Wed, 13 Nov 2024 21:27:28 +0900 Subject: [PATCH 2/2] update field --- .../templates/keypoint_detection/rtmpose_tiny/template.yaml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/otx/tools/templates/keypoint_detection/rtmpose_tiny/template.yaml b/src/otx/tools/templates/keypoint_detection/rtmpose_tiny/template.yaml index 81c2efef1c3..313e3beef5e 100644 --- a/src/otx/tools/templates/keypoint_detection/rtmpose_tiny/template.yaml +++ b/src/otx/tools/templates/keypoint_detection/rtmpose_tiny/template.yaml @@ -25,5 +25,5 @@ gigaflops: 1.3 size: 3.1 # Model spec -model_category: BALANCE +model_category: SPEED is_default_for_task: true