lightmare/ai-voice-cloning
1
0
Fork 0
forked from mrq/ai-voice-cloning
Code Issues Pull Requests Packages Projects Releases Wiki Activity

oops, epoch increments twice

Browse Source
This commit is contained in:
mrq 2023-02-25 15:31:18 +00:00
parent 8b4da29d5f
commit 65329dba31
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/utils.py
View File

@ -542,12 +542,12 @@ class TrainingState():
if line.find('%|') > 0 and not self.open_state: if line.find('%|') > 0 and not self.open_state:
if owner: if owner:
self.open_state = True self.open_state = True
elif lapsed: elif lapsed and self.open_state:
if owner: if owner:
self.open_state = False self.open_state = False
self.epoch = self.epoch + 1 self.epoch = self.epoch + 1
self.it = int(self.epoch * (self.dataset_size / self.batch_size)) self.it = int(self.epoch * (self.dataset_size / self.batch_size))
self.epoch_time_end = time.time() self.epoch_time_end = time.time()
self.epoch_time_delta = self.epoch_time_end-self.epoch_time_start self.epoch_time_delta = self.epoch_time_end-self.epoch_time_start
self.epoch_time_start = time.time() self.epoch_time_start = time.time()