experiments/STT
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

checkpoint_dir always overrides {save,load}_checkpoint_dir

Browse Source
This commit is contained in:
Josh Meyer 2021-07-30 07:57:11 -04:00
parent 979dccdf58
commit 1e79b8703d
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
training/coqui_stt_training/util/config.py
View File

@ -76,10 +76,9 @@ class _SttConfig(Coqpit):
# Checkpoint dir logic #
if self.checkpoint_dir:
if not self.save_checkpoint_dir:
self.save_checkpoint_dir = self.checkpoint_dir
if not self.load_checkpoint_dir:
self.load_checkpoint_dir = self.checkpoint_dir
# checkpoint_dir always overrides {save,load}_checkpoint_dir
self.save_checkpoint_dir = self.checkpoint_dir
self.load_checkpoint_dir = self.checkpoint_dir
else:
if not self.save_checkpoint_dir:
self.save_checkpoint_dir = xdg.save_data_path(