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

Fix CI test for zipformer CTC #1165

Merged
merged 1 commit into from
Jul 5, 2023
Merged

Conversation

csukuangfj
Copy link
Collaborator

See also
https://github.com/k2-fsa/icefall/pull/1130/files#r1252475093

It also fixes the following CI
https://github.com/k2-fsa/icefall/actions/runs/5456594778/jobs/9929614654

2023-07-04 15:59:40,115 INFO [pretrained_ctc.py:293] Number of model parameters: 65802947
Traceback (most recent call last):
  File "./zipformer/pretrained_ctc.py", line 445, in <module>
    main()
  File "/opt/hostedtoolcache/Python/3.8.17/x64/lib/python3.8/site-packages/torch/autograd/grad_mode.py", line 27, in decorate_context
    return func(*args, **kwargs)
  File "./zipformer/pretrained_ctc.py", line 296, in main
    model.load_state_dict(checkpoint["model"], strict=False)
  File "/opt/hostedtoolcache/Python/3.8.17/x64/lib/python3.8/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 AsrModel:
	size mismatch for decoder.embedding.weight: copying a param with shape torch.Size([500, 512]) from checkpoint, the shape in current model is torch.Size([499, 512]).
	size mismatch for joiner.output_linear.weight: copying a param with shape torch.Size([500, 512]) from checkpoint, the shape in current model is torch.Size([499, 512]).

@csukuangfj csukuangfj merged commit 130ad03 into k2-fsa:master Jul 5, 2023
3 checks passed
@csukuangfj csukuangfj deleted the fix-ci-2 branch July 5, 2023 02:38
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

Successfully merging this pull request may close these issues.

1 participant