From ffdfd5f6153280afff5929a75c1fc4321bfdfc91 Mon Sep 17 00:00:00 2001
From: Kohya S <ykumeykume@gmail.com>
Date: Thu, 16 Feb 2023 22:21:36 +0900
Subject: [PATCH] fix name of loss for epoch

---
 train_db.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/train_db.py b/train_db.py
index d36bd8d0..e4f1e54c 100644
--- a/train_db.py
+++ b/train_db.py
@@ -309,7 +309,7 @@ def train(args):
         break
 
     if args.logging_dir is not None:
-      logs = {"epoch_loss": loss_total / len(loss_list)}
+      logs = {"loss/epoch": loss_total / len(loss_list)}
       accelerator.log(logs, step=epoch+1)
 
     accelerator.wait_for_everyone()