NeuralSVB icon indicating copy to clipboard operation
NeuralSVB copied to clipboard

Issue by loading the pre-trained model

Open Alex2025Job opened this issue 2 years ago • 1 comments

Maybe there is an issue by loading the pre-trained model (line 61 ckpt_utils.py)

  • size mismatch for token_embed.weight: copying a param with shape torch.Size([88, 256]) from checkpoint, the shape in current model is torch.Size([87, 256]).
  • size mismatch for asr_decoder.project_out_dim.weight: copying a param with shape torch.Size([88, 256]) from checkpoint, the shape in current model is torch.Size([87, 256])

Alex2025Job avatar Mar 15 '23 17:03 Alex2025Job

I have encountered the same issue and would like to know how to resolve it. Many Thanks!!!

starry0716 avatar Mar 23 '23 02:03 starry0716