tuna2134/sbv2-v2
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Update train_ms.py

Browse Source
This commit is contained in:
Stardust·减
2023-08-23 16:49:33 +08:00
committed by GitHub
parent fe3768b872
commit 432e3923f6
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
train_ms.py
View File

@@ -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}"})