Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

求救!训练自己的数据集时报错,维度问题 #151

Open
dzyanshan opened this issue Nov 2, 2021 · 1 comment
Open

求救!训练自己的数据集时报错,维度问题 #151

dzyanshan opened this issue Nov 2, 2021 · 1 comment

Comments

@dzyanshan
Copy link

RuntimeError: While copying the parameter named ASRN.attentionL2R.char_embeddings, whose dimensions in the model are torch.Size([37, 256]) and whose dimensions in the checkpoint are torch.Size([38, 256]).

@Canjie-Luo
Copy link
Owner

You can train from scratch, or drop the last fully-connected layer in the pretrain model.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants