chore: 添加注释

2026-05-14 04:41:12 +08:00 · 2025-12-15 12:19:57 +08:00 · 2025-12-15 12:19:57 +08:00 · 102e19cefb
commit 102e19cefb
parent bd821aa8e4
1 changed files with 4 additions and 0 deletions
--- 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