search

mkotha / WaveRNN

A WaveRNN implementation
MIT License
198 stars 48 forks source link

Model checkpointing crashes if the folders don't already exist #3

Closed kastnerkyle closed 5 years ago

kastnerkyle commented 5 years ago

I did something like

            model_base_path = str(os.sep).join(paths.model_path().split(os.sep)[:-1])
            step_base_path = str(os.sep).join(paths.step_path().split(os.sep)[:-1])
            hist_base_path = str(os.sep).join(paths.model_hist_path(step).split(os.sep)[:-1])
            if not os.path.exists(model_base_path):
                os.makedirs(model_base_path)
            if not os.path.exists(step_base_path):
                os.makedirs(step_base_path)
            if not os.path.exists(hist_base_path):
                os.makedirs(hist_base_path)

to try and fix this, in models/vqvae.py around line 264

mkotha commented 5 years ago

Oops, you are right, fixed. Thanks.