mrq/ai-voice-cloning
31
74
Fork 43
Code Issues 222 Pull Requests 5 Packages Projects Releases Wiki Activity

keep_training #118

Merged
mrq merged 2 commits from :keep_training into master 2023-03-12 06:47:03 +00:00
Conversation 2 Commits 2 Files Changed 2 +5 -2
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit 29b3d1ae1d - Show all commits

7
src/utils.py
View File

@ -752,8 +752,8 @@ class TrainingState():
models = sorted([ int(d[:-8]) for d in os.listdir(f'{self.dataset_dir}/models/') if d[-8:] == "_gpt.pth" ])
states = sorted([ int(d[:-6]) for d in os.listdir(f'{self.dataset_dir}/training_state/') if d[-6:] == ".state" ])
remove_models = models[:-2]
remove_states = states[:-2]
remove_models = models[:-keep]
remove_states = states[:-keep]
for d in remove_models:
path = f'{self.dataset_dir}/models/{d}_gpt.pth'
@ -898,6 +898,9 @@ class TrainingState():
if should_return:
result = "".join(self.buffer) if not self.training_started else message
if keep_x_past_checkpoints > 0:
self.cleanup_old(keep=keep_x_past_checkpoints)
return (
result,
percent,