forked from mrq/ai-voice-cloning
fix the cleanup actually only doing 2 despite requesting more than 2, surprised no one has pointed it out
This commit is contained in:
parent
8e890d3023
commit
df0edacc60
|
@ -707,8 +707,9 @@ 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'
|
||||
|
|
Loading…
Reference in New Issue
Block a user