[*] restoring model from logs/44k/diffusion\model_0.pt
Traceback (most recent call last):
File "train_diff.py", line 55, in <module>
initial_global_step, model, optimizer = utils.load_model(args.env.expdir, model, optimizer, device=args.device)
File "H:\so-vits-svc\diffusion\logger\utils.py", line 124, in load_model
model.load_state_dict(ckpt['model'], strict=False)
File "H:\so-vits-svc\venv\lib\site-packages\torch\nn\modules\module.py", line 1671, in load_state_dict
raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format(
RuntimeError: Error(s) in loading state_dict for Unit2Mel:
size mismatch for unit_embed.weight: copying a param with shape torch.Size([256, 1024]) from checkpoint, the shape in current model is torch.Size([256, 1280]).
在直接使用从 Diffusion-SVC 下载的预训练模型进行训练的场合,会出现以下错误: