From 102e19cefbb357bcf4c4c901e7e0034b3e0c82a6 Mon Sep 17 00:00:00 2001
From: unanmed <1319491857@qq.com>
Date: Mon, 15 Dec 2025 12:19:57 +0800
Subject: [PATCH] =?UTF-8?q?chore:=20=E6=B7=BB=E5=8A=A0=E6=B3=A8=E9=87=8A?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 ginka/generator/loss.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/ginka/generator/loss.py b/ginka/generator/loss.py
index 83561f4..f4397b2 100644
--- a/ginka/generator/loss.py
+++ b/ginka/generator/loss.py
@@ -408,6 +408,10 @@ class RNNGinkaLoss:
         pass
     
     def rnn_loss(self, fake, target):
+        """
+        fake: [B, C, H, W]
+        target: [B, H, W]
+        """
         weight = torch.ones(self.num_classes)
         weight[0] = 0.3
         weight[1] = 0.5