From 8e6d5afa7b2c819f0197a6015e7e9ceeccf605fe Mon Sep 17 00:00:00 2001 From: sooahleex Date: Thu, 22 Aug 2024 13:54:03 +0900 Subject: [PATCH 1/2] Update step size for eff v2 --- src/otx/algo/classification/timm_model.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/src/otx/algo/classification/timm_model.py b/src/otx/algo/classification/timm_model.py index 4bc3c694c2b..d7e171565a7 100644 --- a/src/otx/algo/classification/timm_model.py +++ b/src/otx/algo/classification/timm_model.py @@ -5,7 +5,8 @@ from __future__ import annotations -from copy import deepcopy +from copy import copy, deepcopy +from math import ceil from typing import TYPE_CHECKING, Literal import torch @@ -267,12 +268,14 @@ def _create_model(self) -> nn.Module: def _build_model(self, head_config: dict) -> nn.Module: backbone = TimmBackbone(backbone=self.backbone, pretrained=self.pretrained) + copied_head_config = copy(head_config) + copied_head_config["step_size"] = (ceil(self.input_size[0] / 32), ceil(self.input_size[1] / 32)) return HLabelClassifier( backbone=backbone, neck=nn.Identity(), head=HierarchicalCBAMClsHead( in_channels=backbone.num_features, - **head_config, + **copied_head_config, ), multiclass_loss=nn.CrossEntropyLoss(), multilabel_loss=AsymmetricAngularLossWithIgnore(gamma_pos=0.0, gamma_neg=1.0, reduction="sum"), From 569ee05444745509889a04429cb82d2a450ca6b7 Mon Sep 17 00:00:00 2001 From: sooahleex Date: Thu, 22 Aug 2024 16:21:20 +0900 Subject: [PATCH 2/2] Update effb0 recipe --- src/otx/recipe/classification/h_label_cls/efficientnet_b0.yaml | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/otx/recipe/classification/h_label_cls/efficientnet_b0.yaml b/src/otx/recipe/classification/h_label_cls/efficientnet_b0.yaml index a734131ad05..d0ea7daec7b 100644 --- a/src/otx/recipe/classification/h_label_cls/efficientnet_b0.yaml +++ b/src/otx/recipe/classification/h_label_cls/efficientnet_b0.yaml @@ -7,6 +7,8 @@ model: class_path: torch.optim.SGD init_args: lr: 0.0049 + momentum: 0.9 + weight_decay: 0.0001 scheduler: class_path: lightning.pytorch.cli.ReduceLROnPlateau