DoctorPopi/ai-voice-cloning-fork
1
0
Fork 0
forked from mrq/ai-voice-cloning
Code Pull Requests Activity

typo

Browse Source
This commit is contained in:
mrq 2023-03-05 20:13:39 +00:00
parent b2e89d8da3
commit b82961ba8a
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/utils.py
View File

@ -760,9 +760,9 @@ class TrainingState():
self.metrics['step'] = [f"{self.epoch}/{self.epochs}"] self.metrics['step'] = [f"{self.epoch}/{self.epochs}"]
if self.epochs != self.its: if self.epochs != self.its:
self.metric.append(f"{self.it}/{self.its}") self.metrics.append(f"{self.it}/{self.its}")
if steps > 1: if steps > 1:
self.metric.append(f"{step}/{steps}") self.metrics.append(f"{step}/{steps}")
self.metrics['step'] = ", ".join(self.metrics['step']) self.metrics['step'] = ", ".join(self.metrics['step'])
if lapsed: if lapsed:
@ -818,7 +818,7 @@ class TrainingState():
deriv = 0 deriv = 0
accum_length = len(self.losses)//2 # i *guess* this is fine when you think about it accum_length = len(self.losses)//2 # i *guess* this is fine when you think about it
loss_value = self.losses[-1]["value"] loss_value = self.losses[-1]["value"]
for i in range(accum_length): for i in range(accum_length):
d1_loss = self.losses[accum_length-i-1]["value"] d1_loss = self.losses[accum_length-i-1]["value"]
d2_loss = self.losses[accum_length-i-2]["value"] d2_loss = self.losses[accum_length-i-2]["value"]