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

fix: typo

Browse Source
This commit is contained in:
unanmed 2026-01-19 00:46:13 +08:00
parent cfc022724a
commit 62ec7daa16
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ginka/train_rnn.py
View File

@ -161,7 +161,7 @@ def train():
val_cond = batch["val_cond"].to(device) val_cond = batch["val_cond"].to(device)
target_map = batch["target_map"].to(device) target_map = batch["target_map"].to(device)
fake_logits, fake_map = ginka_rnn(val_cond, target_map, 1 - gt_prob(epoch, args.epoch)) fake_logits, fake_map = ginka_rnn(val_cond, target_map, 1 - gt_prob(epoch, args.epochs))
val_loss_total += criterion.rnn_loss(fake_logits, target_map).detach() val_loss_total += criterion.rnn_loss(fake_logits, target_map).detach()