diff --git a/train/process_ckpt.py b/train/process_ckpt.py index acc22d8..caf9e05 100644 --- a/train/process_ckpt.py +++ b/train/process_ckpt.py @@ -1,10 +1,13 @@ -import torch, traceback, os, pdb,sys +import torch, traceback, os, pdb, sys + now_dir = os.getcwd() sys.path.append(now_dir) from collections import OrderedDict from i18n import I18nAuto + i18n = I18nAuto() + def savee(ckpt, sr, if_f0, name, epoch, version): try: opt = OrderedDict()