From 66d51f8f33cedeed551f1c2d20952926034a0a9f Mon Sep 17 00:00:00 2001
From: Eunwoo Shin <eunwoo.shin@intel.com>
Date: Fri, 5 Apr 2024 10:40:13 +0900
Subject: [PATCH] Fix a bug occurred when training a RTMdet tiny model (#3266)

change width and hieght order
---
 .../algo/instance_segmentation/heads/custom_rtmdet_ins_head.py  | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/otx/algo/instance_segmentation/heads/custom_rtmdet_ins_head.py b/src/otx/algo/instance_segmentation/heads/custom_rtmdet_ins_head.py
index 948c6a4de3c..580d1a6b931 100644
--- a/src/otx/algo/instance_segmentation/heads/custom_rtmdet_ins_head.py
+++ b/src/otx/algo/instance_segmentation/heads/custom_rtmdet_ins_head.py
@@ -76,8 +76,8 @@ def mask_postprocess(
                 F.interpolate(
                     mask_logits[:, inds],
                     size=[
-                        img_w,
                         img_h,
+                        img_w,
                     ],
                     mode="bilinear",
                     align_corners=False,