hulianyuyy / SEN_CSLR

Self-Emphasizing Network for Continuous Sign Language Recognition (AAAI2023 Oral)
Apache License 2.0
41 stars 2 forks source link

param mismatch error #12

Closed ruoqi77 closed 5 months ago

ruoqi77 commented 5 months ago

Hi, when I run test_one_video.py using CSL-Daily pretrained model (dev_31.10_CSL-Daily.pt), I meet the following error

RuntimeError: Error(s) in loading state_dict for SLRModel: size mismatch for conv1d.fc.weight: copying a param with shape torch.Size([2001, 1024]) from checkpoint, the shape in current model is torch.Size([1296, 1024]). size mismatch for conv1d.fc.bias: copying a param with shape torch.Size([2001]) from checkpoint, the shape in current model is torch.Size([1296]). size mismatch for classifier.weight: copying a param with shape torch.Size([2001, 1024]) from checkpoint, the shape in current model is torch.Size([1296, 1024]). size mismatch for classifier.bias: copying a param with shape torch.Size([2001]) from checkpoint, the shape in current model is torch.Size([1296]).

Should I modify somewhere of the network structure? Thank you!