NeuralSVB icon indicating copy to clipboard operation
NeuralSVB copied to clipboard

Issue by loading the pre-trained model

Open SongLi89 opened this issue 1 year 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])

SongLi89 avatar Mar 15 '23 17:03 SongLi89