diff --git a/external/model-preparation-algorithm/configs/segmentation/ocr-lite-hrnet-18/hpo_config.yaml b/external/model-preparation-algorithm/configs/segmentation/ocr-lite-hrnet-18/hpo_config.yaml new file mode 100644 index 00000000000..819a3ac0368 --- /dev/null +++ b/external/model-preparation-algorithm/configs/segmentation/ocr-lite-hrnet-18/hpo_config.yaml @@ -0,0 +1,15 @@ +metric: mDice +search_algorithm: asha +hp_space: + learning_parameters.learning_rate: + param_type: qloguniform + range: + - 0.0002 + - 0.005 + - 0.0001 + learning_parameters.batch_size: + param_type: qloguniform + range: + - 6 + - 12 + - 2 diff --git a/external/model-preparation-algorithm/configs/segmentation/ocr-lite-hrnet-18/model.py b/external/model-preparation-algorithm/configs/segmentation/ocr-lite-hrnet-18/model.py new file mode 100644 index 00000000000..8a162147228 --- /dev/null +++ b/external/model-preparation-algorithm/configs/segmentation/ocr-lite-hrnet-18/model.py @@ -0,0 +1,6 @@ +_base_ = [ + '../../../submodule/models/segmentation/ocr_litehrnet18_mod2.yaml', +] + +load_from = 'https://storage.openvinotoolkit.org/repositories/openvino_training_extensions/models/custom_semantic_segmentation/litehrnet18_imagenet1k_rsc.pth' +fp16 = dict(loss_scale=512.) diff --git a/external/model-preparation-algorithm/configs/segmentation/ocr-lite-hrnet-18/template.yaml b/external/model-preparation-algorithm/configs/segmentation/ocr-lite-hrnet-18/template.yaml new file mode 100644 index 00000000000..4763d1d219d --- /dev/null +++ b/external/model-preparation-algorithm/configs/segmentation/ocr-lite-hrnet-18/template.yaml @@ -0,0 +1,52 @@ +# Description. +model_template_id: Custom_Semantic_Segmentation_Lite-HRNet-18_OCR +name: Lite-HRNet-18 +task_type: SEGMENTATION +task_family: VISION +instantiation: "CLASS" +summary: Class-Incremental Semantic Segmentation with middle-sized architecture which based on the Lite-HRNet backbone for the balance between the fast inference and long training. (deprecated in next version) +application: ~ + +# Algo backend. +framework: OTESegmentation v0.14.0 + +# Task implementations. +entrypoints: + base: mpa_tasks.apis.segmentation.SegmentationTrainTask + openvino: segmentation_tasks.apis.segmentation.OpenVINOSegmentationTask + nncf: mpa_tasks.apis.segmentation.SegmentationNNCFTask +base_model_path: ../../../../mmsegmentation/configs/custom-sematic-segmentation/ocr-lite-hrnet-18/template_experimental.yaml + +# Capabilities. +capabilities: + - compute_representations + +# Hyperparameters. +hyper_parameters: + base_path: ../configuration.yaml + parameter_overrides: + learning_parameters: + batch_size: + default_value: 8 + learning_rate: + default_value: 0.001 + auto_hpo_state: POSSIBLE + learning_rate_fixed_iters: + default_value: 0 + learning_rate_warmup_iters: + default_value: 100 + num_iters: + default_value: 300 + algo_backend: + train_type: + default_value: Incremental + +# Training resources. +max_nodes: 1 +training_targets: + - GPU + - CPU + +# Stats. +gigaflops: 3.45 +size: 4.5