From ddc9e2f4bb4e27570c72dfda788947b6f69624e8 Mon Sep 17 00:00:00 2001
From: "copilot-swe-agent[bot]" <198982749+Copilot@users.noreply.github.com>
Date: Wed, 22 Apr 2026 08:28:51 +0000
Subject: [PATCH] Fix fallback decoder config handling

Agent-Logs-Url: https://github.com/IBB666/IBB_POSE/sessions/be2c1a27-aacc-43a8-adc6-0bb4dbae296e

Co-authored-by: kukulu00 <1170651+kukulu00@users.noreply.github.com>
---
 models/HeatmapHead.py | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/models/HeatmapHead.py b/models/HeatmapHead.py
index 57bc725..14ce827 100644
--- a/models/HeatmapHead.py
+++ b/models/HeatmapHead.py
@@ -13,7 +13,6 @@ import warnings
 from types import SimpleNamespace
 
 import numpy as np
-import torch
 from torch import nn
 
 try:
@@ -125,7 +124,15 @@ class _FallbackHeatmapHead(nn.Module):
         self.in_channels = in_channels
         self.out_channels = out_channels
         self.loss_module = None
-        self.decoder = _FallbackUDPHeatmap(**decoder) if decoder is not None else None
+        self.decoder = (
+            _FallbackUDPHeatmap(
+                input_size=decoder["input_size"],
+                heatmap_size=decoder["heatmap_size"],
+                sigma=decoder.get("sigma", 2.0),
+            )
+            if decoder is not None
+            else None
+        )
 
         if deconv_out_channels:
             self.deconv_layers = self._make_deconv_layers(