#remove .pyt extension and step number
prev_model_name = re.sub(
r'_[0-9]+$', '', re.sub(r'\.pyt$', '',
os.path.basename(args.load)))
prev_model_basename = prev_model_name.split('_')[0]
model_basename = model_name.split('_')[0]
if prev_model_basename != model_basename and not args.force:
sys.exit(
f'refusing to load {args.load} because its prev_model_basename:({prev_model_basename}) is not model_basename:({model_basename})'
)
if args.generate:
paths = env.Paths(prev_model_name, data_path)
prev_path = args.load
else:
prev_path = paths.model_path()
step, epoch = env.restore(prev_path, model, optimiser)
#model.freeze_encoder()
if args.generate:
if args.tokens_path:
model.do_generate_from_tokens(paths, args.tokens_path, verbose=True)
else:
model.do_generate(paths,
data_path,
index,
args.test_speakers,
args.test_utts_per_speaker,
use_half=use_half,
verbose=True,
only_discrete=args.only_gen_discrete)
if args.load:
prev_model_name = re.sub(
r'_[0-9]+$', '', re.sub(r'\.pyt$', '',
os.path.basename(args.load)))
prev_model_basename = prev_model_name.split('_')[0]
model_basename = model_name.split('_')[0]
if prev_model_basename != model_basename and not args.force:
sys.exit(
f'refusing to load {args.load} because its basename ({prev_model_basename}) is not {model_basename}'
)
if args.generate:
paths = env.Paths(prev_model_name, data_path)
prev_path = args.load
else:
prev_path = paths.model_path()
step = env.restore(prev_path, model)
#model.freeze_encoder()
optimiser = optim.AdamW(model.parameters(),
betas=(0.9, 0.999),
weight_decay=0.01)
if args.generate:
model.do_generate(paths,
step,
data_path,
test_index,
use_half=use_half,
verbose=True) #, deterministic=True)
else:
logger.set_logfile(paths.logfile_path())