AncTe/ginka-generator
1
0
Fork 0
mirror of https://github.com/unanmed/ginka-generator.git synced 2026-05-23 12:21:11 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: 验证报错

Browse Source
This commit is contained in:
unanmed 2026-02-06 16:51:18 +08:00
parent dc6d1c69be
commit cf1cbbc680
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
ginka/train_vae.py
View File

@ -150,9 +150,8 @@ def train():
loss, reco_loss, kl_loss = criterion.vae_loss(fake_logits, target_map, mu, logvar, KL_BETA) loss, reco_loss, kl_loss = criterion.vae_loss(fake_logits, target_map, mu, logvar, KL_BETA)
val_loss_total += loss.detach() val_loss_total += loss.detach()
idx += 1
avg_loss_val = val_loss_total.item() / len(dataloader_val)
if avg_loss_val < 0.5 and gt_prob > 0: if avg_loss_val < 0.5 and gt_prob > 0:
gt_prob -= 0.01 gt_prob -= 0.01