Fix dir_exits error

This commit is contained in:
litagin02
2024-01-12 13:43:29 +09:00
parent 1067b8ec60
commit a8e0619ad4
2 changed files with 2 additions and 1 deletions

View File

@@ -49,7 +49,7 @@ def get_bert_feature(
style_res = torch.cat(style_res["hidden_states"][-3:-2], -1)[0].cpu()
style_res_mean = style_res.mean(0)
assert len(word2ph) == len(text) + 2
assert len(word2ph) == len(text) + 2, text
word2phone = word2ph
phone_level_feature = []
for i in range(len(word2phone)):

View File

@@ -65,6 +65,7 @@ def initialize(model_name, batch_size, epochs, save_every_steps, bf16_run):
shutil.copytree(
src=model_path,
dst=os.path.join(dataset_path, "models_backup"),
dirs_exist_ok=True,
)
shutil.rmtree(model_path)
try: