AncTe/ginka-generator
1
0
Fork 0
mirror of https://github.com/unanmed/ginka-generator.git synced 2026-05-20 17:22:41 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

chore: 添加注释

Browse Source
This commit is contained in:
unanmed 2025-12-15 12:19:57 +08:00
parent bd821aa8e4
commit 102e19cefb
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
ginka/generator/loss.py
View File

@ -408,6 +408,10 @@ class RNNGinkaLoss:
pass pass
def rnn_loss(self, fake, target): def rnn_loss(self, fake, target):
"""
fake: [B, C, H, W]
target: [B, H, W]
"""
weight = torch.ones(self.num_classes) weight = torch.ones(self.num_classes)
weight[0] = 0.3 weight[0] = 0.3
weight[1] = 0.5 weight[1] = 0.5