From 35225a35daa5b346a132629a2efc0b1a0a7de067 Mon Sep 17 00:00:00 2001 From: mrq Date: Sun, 5 Mar 2023 14:19:41 +0000 Subject: [PATCH] oops v2 --- dlas | 2 +- src/utils.py | 7 +++++-- tortoise-tts | 2 +- 3 files changed, 7 insertions(+), 4 deletions(-) diff --git a/dlas b/dlas index 71cc43e..6eb7ebf 160000 --- a/dlas +++ b/dlas @@ -1 +1 @@ -Subproject commit 71cc43e65cd47c6704d20c99006a3e78feb2400d +Subproject commit 6eb7ebf847cf2e4761536391de841dc4209d1e63 diff --git a/src/utils.py b/src/utils.py index 3552ea9..b1e212c 100755 --- a/src/utils.py +++ b/src/utils.py @@ -752,7 +752,7 @@ class TrainingState(): self.it_taken = self.it_taken + 1 if self.it_time_delta: try: - rate = f'{"{:.3f}".format(self.it_time_delta)}s/it' if self.it_time_delta >= 1 else f'{"{:.3f}".format(1/self.it_time_delta)}it/s' + rate = f'{"{:.3f}".format(self.it_time_delta)}s/it' if self.it_time_delta >= 1 or self.it_time_delta == 0 else f'{"{:.3f}".format(1/self.it_time_delta)}it/s' self.it_rate = rate except Exception as e: pass @@ -767,7 +767,10 @@ class TrainingState(): self.epoch_time_end = time.time() self.epoch_time_delta = self.epoch_time_end-self.epoch_time_start self.epoch_time_start = time.time() - self.epoch_rate = f'{"{:.3f}".format(self.epoch_time_delta)}s/epoch' if self.epoch_time_delta >= 1 else f'{"{:.3f}".format(1/self.epoch_time_delta)}epoch/s' # I doubt anyone will have it/s rates, but its here + try: + self.epoch_rate = f'{"{:.3f}".format(self.epoch_time_delta)}s/epoch' if self.epoch_time_delta >= 1 or self.epoch_time_delta == 0 else f'{"{:.3f}".format(1/self.epoch_time_delta)}epoch/s' # I doubt anyone will have it/s rates, but its here + except Exception as e: + pass #self.eta = (self.epochs - self.epoch) * self.epoch_time_delta self.epoch_time_deltas = self.epoch_time_deltas + self.epoch_time_delta diff --git a/tortoise-tts b/tortoise-tts index 06bdf72..0f3261e 160000 --- a/tortoise-tts +++ b/tortoise-tts @@ -1 +1 @@ -Subproject commit 06bdf72b89af78609059a9e0bc87a1ac88bb1c44 +Subproject commit 0f3261e0717be9e3a59b652fea7c1a388c8fa43a