From cf1cbbc680c23c3e38a76a20ae3898a3608b8371 Mon Sep 17 00:00:00 2001
From: unanmed <1319491857@qq.com>
Date: Fri, 6 Feb 2026 16:51:18 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E9=AA=8C=E8=AF=81=E6=8A=A5=E9=94=99?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 ginka/train_vae.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/ginka/train_vae.py b/ginka/train_vae.py
index a13adcb..c3d4b65 100644
--- a/ginka/train_vae.py
+++ b/ginka/train_vae.py
@@ -150,9 +150,8 @@ def train():
 
                 loss, reco_loss, kl_loss = criterion.vae_loss(fake_logits, target_map, mu, logvar, KL_BETA)
                 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:
             gt_prob -= 0.01