From 432e3923f6cfe217d9d153eb0355438aab673852 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Stardust=C2=B7=E5=87=8F?= <2225664821@qq.com>
Date: Wed, 23 Aug 2023 16:49:33 +0800
Subject: [PATCH] Update train_ms.py

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

diff --git a/train_ms.py b/train_ms.py
index 2eb80e3..3e0a187 100644
--- a/train_ms.py
+++ b/train_ms.py
@@ -303,7 +303,7 @@ def train_and_evaluate(rank, epoch, hps, nets, optims, schedulers, scaler, loade
                 scalar_dict.update({"loss/d_r/{}".format(i): v for i, v in enumerate(losses_disc_r)})
                 scalar_dict.update({"loss/d_g/{}".format(i): v for i, v in enumerate(losses_disc_g)})
                 if net_dur_disc is not None:
-                    scalar_dict.update({"loss/dur_disc_r"  f"{losses_dur_disc_r}"})
+                    scalar_dict.update({"loss/dur_disc_r" : f"{losses_dur_disc_r}"})
                     scalar_dict.update({"loss/dur_disc_g" : f"{losses_dur_disc_g}"})
                     scalar_dict.update({"loss/dur_gen" : f"{loss_dur_gen}"})