diff --git a/configs/ljs_li44_tmbert_ft_small_s1_arpa.json b/configs/ljs_li44_tmbert_ft_small_s1_arpa.json index d42ee5f..fc7d05b 100644 --- a/configs/ljs_li44_tmbert_ft_small_s1_arpa.json +++ b/configs/ljs_li44_tmbert_ft_small_s1_arpa.json @@ -1,7 +1,7 @@ { "train": { "log_interval": 200, - "eval_interval": 1000, + "eval_interval": 1500, "save_interval": 2000, "seed": 1234, "epochs": 20000, diff --git a/preprocess.py b/preprocess.py index f53cef3..efb740b 100644 --- a/preprocess.py +++ b/preprocess.py @@ -42,15 +42,15 @@ if __name__ == '__main__': for i in tqdm(range(len(filepaths_and_text))): original_text = filepaths_and_text[i][args.text_index] - original_fname = filepaths_and_text[i][0].split(".")[0] # wavs/cat.wav -> wavs/cat + original_fname = filepaths_and_text[i][0] - moji_filename = original_fname + ".torchmoji" + moji_filename = original_fname.replace(".wav",".torchmoji") if not os.path.isfile(moji_filename): moji_result = moji(original_text) torch.save(moji_result, moji_filename) - bert_filename = original_fname + ".bert" + bert_filename = original_fname.replace(".wav",".bert") if not os.path.isfile(bert_filename): bert_res, _ = bert_f.infer(original_text) torch.save(bert_res,bert_filename)