运行代码报错：oad RGB pretrained model from: /data2/local_userdata/heyuting/REUTrack/CEUTrack/lib/models/ceutrack/pretrained_models/mae_pretrain_vit_base.pth
Traceback (most recent call last):
  File "/data2/local_userdata/heyuting/REUTrack/CEUTrack/lib/train/run_training.py", line 110, in <module>
    main()
  File "/data2/local_userdata/heyuting/REUTrack/CEUTrack/lib/train/run_training.py", line 101, in main
    run_training(args.script, args.config, cudnn_benchmark=args.cudnn_benchmark,
  File "/data2/local_userdata/heyuting/REUTrack/CEUTrack/lib/train/run_training.py", line 75, in run_training
    expr_func(settings)
  File "/data2/local_userdata/heyuting/REUTrack/CEUTrack/lib/train/../../lib/train/train_script.py", line 55, in run
    net = build_ceutrack(cfg)
  File "/data2/local_userdata/heyuting/REUTrack/CEUTrack/lib/train/../../lib/models/ceutrack/ceutrack.py", line 128, in build_ceutrack
    backbone = vit_base_patch16_224_ce(pretrained, rgb_pretrain=rgb_pretrain, event_pretrain=event_pretrain,
  File "/data2/local_userdata/heyuting/REUTrack/CEUTrack/lib/train/../../lib/models/ceutrack/vit_ce.py", line 217, in vit_base_patch16_224_ce
    model = _create_vision_transformer(pretrained=pretrained, rgb_pretrain=rgb_pretrain, event_pretrain=event_pretrain, **model_kwargs)
  File "/data2/local_userdata/heyuting/REUTrack/CEUTrack/lib/train/../../lib/models/ceutrack/vit_ce.py", line 198, in _create_vision_transformer
    model = VisionTransformerCE(rgb_pretrain=rgb_pretrain, event_pretrain=event_pretrain, **kwargs)
  File "/data2/local_userdata/heyuting/REUTrack/CEUTrack/lib/train/../../lib/models/ceutrack/vit_ce.py", line 105, in __init__
    self.event_transformer.load_state_dict(checkpoint['model'], strict=False)
  File "/home/heyuting/anaconda3/envs/mamba/lib/python3.10/site-packages/torch/nn/modules/module.py", line 2152, in load_state_dict
    raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format(
RuntimeError: Error(s) in loading state_dict for VisionMamba:
        size mismatch for patch_embed.proj.weight: copying a param with shape torch.Size([384, 3, 16, 16]) from checkpoint, the shape in current model is torch.Size([768, 3, 16, 16]).
        size mismatch for patch_embed.proj.bias: copying a param with shape torch.Size([384]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for layers.0.mixer.A_log: copying a param with shape torch.Size([768, 16]) from checkpoint, the shape in current model is torch.Size([1536, 16]).
        size mismatch for layers.0.mixer.D: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.0.mixer.in_proj.weight: copying a param with shape torch.Size([1536, 384]) from checkpoint, the shape in current model is torch.Size([3072, 768]).
        size mismatch for layers.0.mixer.conv1d.weight: copying a param with shape torch.Size([768, 1, 4]) from checkpoint, the shape in current model is torch.Size([1536, 1, 4]).
        size mismatch for layers.0.mixer.conv1d.bias: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.0.mixer.x_proj.weight: copying a param with shape torch.Size([56, 768]) from checkpoint, the shape in current model is torch.Size([80, 1536]).
        size mismatch for layers.0.mixer.dt_proj.weight: copying a param with shape torch.Size([768, 24]) from checkpoint, the shape in current model is torch.Size([1536, 48]).
        size mismatch for layers.0.mixer.dt_proj.bias: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.0.mixer.out_proj.weight: copying a param with shape torch.Size([384, 768]) from checkpoint, the shape in current model is torch.Size([768, 1536]).
        size mismatch for layers.0.norm.weight: copying a param with shape torch.Size([384]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for layers.1.mixer.A_log: copying a param with shape torch.Size([768, 16]) from checkpoint, the shape in current model is torch.Size([1536, 16]).
        size mismatch for layers.1.mixer.D: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.1.mixer.in_proj.weight: copying a param with shape torch.Size([1536, 384]) from checkpoint, the shape in current model is torch.Size([3072, 768]).
        size mismatch for layers.1.mixer.conv1d.weight: copying a param with shape torch.Size([768, 1, 4]) from checkpoint, the shape in current model is torch.Size([1536, 1, 4]).
        size mismatch for layers.1.mixer.conv1d.bias: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.1.mixer.x_proj.weight: copying a param with shape torch.Size([56, 768]) from checkpoint, the shape in current model is torch.Size([80, 1536]).
        size mismatch for layers.1.mixer.dt_proj.weight: copying a param with shape torch.Size([768, 24]) from checkpoint, the shape in current model is torch.Size([1536, 48]).
        size mismatch for layers.1.mixer.dt_proj.bias: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.1.mixer.out_proj.weight: copying a param with shape torch.Size([384, 768]) from checkpoint, the shape in current model is torch.Size([768, 1536]).
        size mismatch for layers.1.norm.weight: copying a param with shape torch.Size([384]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for layers.2.mixer.A_log: copying a param with shape torch.Size([768, 16]) from checkpoint, the shape in current model is torch.Size([1536, 16]).
        size mismatch for layers.2.mixer.D: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.2.mixer.in_proj.weight: copying a param with shape torch.Size([1536, 384]) from checkpoint, the shape in current model is torch.Size([3072, 768]).
        size mismatch for layers.2.mixer.conv1d.weight: copying a param with shape torch.Size([768, 1, 4]) from checkpoint, the shape in current model is torch.Size([1536, 1, 4]).
        size mismatch for layers.2.mixer.conv1d.bias: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.2.mixer.x_proj.weight: copying a param with shape torch.Size([56, 768]) from checkpoint, the shape in current model is torch.Size([80, 1536]).
        size mismatch for layers.2.mixer.dt_proj.weight: copying a param with shape torch.Size([768, 24]) from checkpoint, the shape in current model is torch.Size([1536, 48]).
        size mismatch for layers.2.mixer.dt_proj.bias: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.2.mixer.out_proj.weight: copying a param with shape torch.Size([384, 768]) from checkpoint, the shape in current model is torch.Size([768, 1536]).
        size mismatch for layers.2.norm.weight: copying a param with shape torch.Size([384]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for layers.3.mixer.A_log: copying a param with shape torch.Size([768, 16]) from checkpoint, the shape in current model is torch.Size([1536, 16]).
        size mismatch for layers.3.mixer.D: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.3.mixer.in_proj.weight: copying a param with shape torch.Size([1536, 384]) from checkpoint, the shape in current model is torch.Size([3072, 768]).
        size mismatch for layers.3.mixer.conv1d.weight: copying a param with shape torch.Size([768, 1, 4]) from checkpoint, the shape in current model is torch.Size([1536, 1, 4]).
        size mismatch for layers.3.mixer.conv1d.bias: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.3.mixer.x_proj.weight: copying a param with shape torch.Size([56, 768]) from checkpoint, the shape in current model is torch.Size([80, 1536]).
        size mismatch for layers.3.mixer.dt_proj.weight: copying a param with shape torch.Size([768, 24]) from checkpoint, the shape in current model is torch.Size([1536, 48]).
        size mismatch for layers.3.mixer.dt_proj.bias: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.3.mixer.out_proj.weight: copying a param with shape torch.Size([384, 768]) from checkpoint, the shape in current model is torch.Size([768, 1536]).
        size mismatch for layers.3.norm.weight: copying a param with shape torch.Size([384]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for layers.4.mixer.A_log: copying a param with shape torch.Size([768, 16]) from checkpoint, the shape in current model is torch.Size([1536, 16]).
        size mismatch for layers.4.mixer.D: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.4.mixer.in_proj.weight: copying a param with shape torch.Size([1536, 384]) from checkpoint, the shape in current model is torch.Size([3072, 768]).
        size mismatch for layers.4.mixer.conv1d.weight: copying a param with shape torch.Size([768, 1, 4]) from checkpoint, the shape in current model is torch.Size([1536, 1, 4]).
        size mismatch for layers.4.mixer.conv1d.bias: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.4.mixer.x_proj.weight: copying a param with shape torch.Size([56, 768]) from checkpoint, the shape in current model is torch.Size([80, 1536]).
        size mismatch for layers.4.mixer.dt_proj.weight: copying a param with shape torch.Size([768, 24]) from checkpoint, the shape in current model is torch.Size([1536, 48]).
        size mismatch for layers.4.mixer.dt_proj.bias: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.4.mixer.out_proj.weight: copying a param with shape torch.Size([384, 768]) from checkpoint, the shape in current model is torch.Size([768, 1536]).
        size mismatch for layers.4.norm.weight: copying a param with shape torch.Size([384]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for layers.5.mixer.A_log: copying a param with shape torch.Size([768, 16]) from checkpoint, the shape in current model is torch.Size([1536, 16]).
        size mismatch for layers.5.mixer.D: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.5.mixer.in_proj.weight: copying a param with shape torch.Size([1536, 384]) from checkpoint, the shape in current model is torch.Size([3072, 768]).
        size mismatch for layers.5.mixer.conv1d.weight: copying a param with shape torch.Size([768, 1, 4]) from checkpoint, the shape in current model is torch.Size([1536, 1, 4]).
        size mismatch for layers.5.mixer.conv1d.bias: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.5.mixer.x_proj.weight: copying a param with shape torch.Size([56, 768]) from checkpoint, the shape in current model is torch.Size([80, 1536]).
        size mismatch for layers.5.mixer.dt_proj.weight: copying a param with shape torch.Size([768, 24]) from checkpoint, the shape in current model is torch.Size([1536, 48]).
        size mismatch for layers.5.mixer.dt_proj.bias: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.5.mixer.out_proj.weight: copying a param with shape torch.Size([384, 768]) from checkpoint, the shape in current model is torch.Size([768, 1536]).
        size mismatch for layers.5.norm.weight: copying a param with shape torch.Size([384]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for layers.6.mixer.A_log: copying a param with shape torch.Size([768, 16]) from checkpoint, the shape in current model is torch.Size([1536, 16]).
        size mismatch for layers.6.mixer.D: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.6.mixer.in_proj.weight: copying a param with shape torch.Size([1536, 384]) from checkpoint, the shape in current model is torch.Size([3072, 768]).
        size mismatch for layers.6.mixer.conv1d.weight: copying a param with shape torch.Size([768, 1, 4]) from checkpoint, the shape in current model is torch.Size([1536, 1, 4]).
        size mismatch for layers.6.mixer.conv1d.bias: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.6.mixer.x_proj.weight: copying a param with shape torch.Size([56, 768]) from checkpoint, the shape in current model is torch.Size([80, 1536]).
        size mismatch for layers.6.mixer.dt_proj.weight: copying a param with shape torch.Size([768, 24]) from checkpoint, the shape in current model is torch.Size([1536, 48]).
        size mismatch for layers.6.mixer.dt_proj.bias: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.6.mixer.out_proj.weight: copying a param with shape torch.Size([384, 768]) from checkpoint, the shape in current model is torch.Size([768, 1536]).
        size mismatch for layers.6.norm.weight: copying a param with shape torch.Size([384]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for layers.7.mixer.A_log: copying a param with shape torch.Size([768, 16]) from checkpoint, the shape in current model is torch.Size([1536, 16]).
        size mismatch for layers.7.mixer.D: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.7.mixer.in_proj.weight: copying a param with shape torch.Size([1536, 384]) from checkpoint, the shape in current model is torch.Size([3072, 768]).
        size mismatch for layers.7.mixer.conv1d.weight: copying a param with shape torch.Size([768, 1, 4]) from checkpoint, the shape in current model is torch.Size([1536, 1, 4]).
        size mismatch for layers.7.mixer.conv1d.bias: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.7.mixer.x_proj.weight: copying a param with shape torch.Size([56, 768]) from checkpoint, the shape in current model is torch.Size([80, 1536]).
        size mismatch for layers.7.mixer.dt_proj.weight: copying a param with shape torch.Size([768, 24]) from checkpoint, the shape in current model is torch.Size([1536, 48]).
        size mismatch for layers.7.mixer.dt_proj.bias: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.7.mixer.out_proj.weight: copying a param with shape torch.Size([384, 768]) from checkpoint, the shape in current model is torch.Size([768, 1536]).
        size mismatch for layers.7.norm.weight: copying a param with shape torch.Size([384]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for layers.8.mixer.A_log: copying a param with shape torch.Size([768, 16]) from checkpoint, the shape in current model is torch.Size([1536, 16]).
        size mismatch for layers.8.mixer.D: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.8.mixer.in_proj.weight: copying a param with shape torch.Size([1536, 384]) from checkpoint, the shape in current model is torch.Size([3072, 768]).
        size mismatch for layers.8.mixer.conv1d.weight: copying a param with shape torch.Size([768, 1, 4]) from checkpoint, the shape in current model is torch.Size([1536, 1, 4]).
        size mismatch for layers.8.mixer.conv1d.bias: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.8.mixer.x_proj.weight: copying a param with shape torch.Size([56, 768]) from checkpoint, the shape in current model is torch.Size([80, 1536]).
        size mismatch for layers.8.mixer.dt_proj.weight: copying a param with shape torch.Size([768, 24]) from checkpoint, the shape in current model is torch.Size([1536, 48]).
        size mismatch for layers.8.mixer.dt_proj.bias: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.8.mixer.out_proj.weight: copying a param with shape torch.Size([384, 768]) from checkpoint, the shape in current model is torch.Size([768, 1536]).
        size mismatch for layers.8.norm.weight: copying a param with shape torch.Size([384]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for layers.9.mixer.A_log: copying a param with shape torch.Size([768, 16]) from checkpoint, the shape in current model is torch.Size([1536, 16]).
        size mismatch for layers.9.mixer.D: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.9.mixer.in_proj.weight: copying a param with shape torch.Size([1536, 384]) from checkpoint, the shape in current model is torch.Size([3072, 768]).
        size mismatch for layers.9.mixer.conv1d.weight: copying a param with shape torch.Size([768, 1, 4]) from checkpoint, the shape in current model is torch.Size([1536, 1, 4]).
        size mismatch for layers.9.mixer.conv1d.bias: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.9.mixer.x_proj.weight: copying a param with shape torch.Size([56, 768]) from checkpoint, the shape in current model is torch.Size([80, 1536]).
        size mismatch for layers.9.mixer.dt_proj.weight: copying a param with shape torch.Size([768, 24]) from checkpoint, the shape in current model is torch.Size([1536, 48]).
        size mismatch for layers.9.mixer.dt_proj.bias: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.9.mixer.out_proj.weight: copying a param with shape torch.Size([384, 768]) from checkpoint, the shape in current model is torch.Size([768, 1536]).
        size mismatch for layers.9.norm.weight: copying a param with shape torch.Size([384]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for layers.10.mixer.A_log: copying a param with shape torch.Size([768, 16]) from checkpoint, the shape in current model is torch.Size([1536, 16]).
        size mismatch for layers.10.mixer.D: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.10.mixer.in_proj.weight: copying a param with shape torch.Size([1536, 384]) from checkpoint, the shape in current model is torch.Size([3072, 768]).
        size mismatch for layers.10.mixer.conv1d.weight: copying a param with shape torch.Size([768, 1, 4]) from checkpoint, the shape in current model is torch.Size([1536, 1, 4]).
        size mismatch for layers.10.mixer.conv1d.bias: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.10.mixer.x_proj.weight: copying a param with shape torch.Size([56, 768]) from checkpoint, the shape in current model is torch.Size([80, 1536]).
        size mismatch for layers.10.mixer.dt_proj.weight: copying a param with shape torch.Size([768, 24]) from checkpoint, the shape in current model is torch.Size([1536, 48]).
        size mismatch for layers.10.mixer.dt_proj.bias: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.10.mixer.out_proj.weight: copying a param with shape torch.Size([384, 768]) from checkpoint, the shape in current model is torch.Size([768, 1536]).
        size mismatch for layers.10.norm.weight: copying a param with shape torch.Size([384]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for layers.11.mixer.A_log: copying a param with shape torch.Size([768, 16]) from checkpoint, the shape in current model is torch.Size([1536, 16]).
        size mismatch for layers.11.mixer.D: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.11.mixer.in_proj.weight: copying a param with shape torch.Size([1536, 384]) from checkpoint, the shape in current model is torch.Size([3072, 768]).
        size mismatch for layers.11.mixer.conv1d.weight: copying a param with shape torch.Size([768, 1, 4]) from checkpoint, the shape in current model is torch.Size([1536, 1, 4]).
        size mismatch for layers.11.mixer.conv1d.bias: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.11.mixer.x_proj.weight: copying a param with shape torch.Size([56, 768]) from checkpoint, the shape in current model is torch.Size([80, 1536]).
        size mismatch for layers.11.mixer.dt_proj.weight: copying a param with shape torch.Size([768, 24]) from checkpoint, the shape in current model is torch.Size([1536, 48]).
        size mismatch for layers.11.mixer.dt_proj.bias: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.11.mixer.out_proj.weight: copying a param with shape torch.Size([384, 768]) from checkpoint, the shape in current model is torch.Size([768, 1536]).
        size mismatch for layers.11.norm.weight: copying a param with shape torch.Size([384]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for layers.12.mixer.A_log: copying a param with shape torch.Size([768, 16]) from checkpoint, the shape in current model is torch.Size([1536, 16]).
        size mismatch for layers.12.mixer.D: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.12.mixer.in_proj.weight: copying a param with shape torch.Size([1536, 384]) from checkpoint, the shape in current model is torch.Size([3072, 768]).
        size mismatch for layers.12.mixer.conv1d.weight: copying a param with shape torch.Size([768, 1, 4]) from checkpoint, the shape in current model is torch.Size([1536, 1, 4]).
        size mismatch for layers.12.mixer.conv1d.bias: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.12.mixer.x_proj.weight: copying a param with shape torch.Size([56, 768]) from checkpoint, the shape in current model is torch.Size([80, 1536]).
        size mismatch for layers.12.mixer.dt_proj.weight: copying a param with shape torch.Size([768, 24]) from checkpoint, the shape in current model is torch.Size([1536, 48]).
        size mismatch for layers.12.mixer.dt_proj.bias: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.12.mixer.out_proj.weight: copying a param with shape torch.Size([384, 768]) from checkpoint, the shape in current model is torch.Size([768, 1536]).
        size mismatch for layers.12.norm.weight: copying a param with shape torch.Size([384]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for layers.13.mixer.A_log: copying a param with shape torch.Size([768, 16]) from checkpoint, the shape in current model is torch.Size([1536, 16]).
        size mismatch for layers.13.mixer.D: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.13.mixer.in_proj.weight: copying a param with shape torch.Size([1536, 384]) from checkpoint, the shape in current model is torch.Size([3072, 768]).
        size mismatch for layers.13.mixer.conv1d.weight: copying a param with shape torch.Size([768, 1, 4]) from checkpoint, the shape in current model is torch.Size([1536, 1, 4]).
        size mismatch for layers.13.mixer.conv1d.bias: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.13.mixer.x_proj.weight: copying a param with shape torch.Size([56, 768]) from checkpoint, the shape in current model is torch.Size([80, 1536]).
        size mismatch for layers.13.mixer.dt_proj.weight: copying a param with shape torch.Size([768, 24]) from checkpoint, the shape in current model is torch.Size([1536, 48]).
        size mismatch for layers.13.mixer.dt_proj.bias: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.13.mixer.out_proj.weight: copying a param with shape torch.Size([384, 768]) from checkpoint, the shape in current model is torch.Size([768, 1536]).
        size mismatch for layers.13.norm.weight: copying a param with shape torch.Size([384]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for layers.14.mixer.A_log: copying a param with shape torch.Size([768, 16]) from checkpoint, the shape in current model is torch.Size([1536, 16]).
        size mismatch for layers.14.mixer.D: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.14.mixer.in_proj.weight: copying a param with shape torch.Size([1536, 384]) from checkpoint, the shape in current model is torch.Size([3072, 768]).
        size mismatch for layers.14.mixer.conv1d.weight: copying a param with shape torch.Size([768, 1, 4]) from checkpoint, the shape in current model is torch.Size([1536, 1, 4]).
        size mismatch for layers.14.mixer.conv1d.bias: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.14.mixer.x_proj.weight: copying a param with shape torch.Size([56, 768]) from checkpoint, the shape in current model is torch.Size([80, 1536]).
        size mismatch for layers.14.mixer.dt_proj.weight: copying a param with shape torch.Size([768, 24]) from checkpoint, the shape in current model is torch.Size([1536, 48]).
        size mismatch for layers.14.mixer.dt_proj.bias: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.14.mixer.out_proj.weight: copying a param with shape torch.Size([384, 768]) from checkpoint, the shape in current model is torch.Size([768, 1536]).
        size mismatch for layers.14.norm.weight: copying a param with shape torch.Size([384]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for layers.15.mixer.A_log: copying a param with shape torch.Size([768, 16]) from checkpoint, the shape in current model is torch.Size([1536, 16]).
        size mismatch for layers.15.mixer.D: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.15.mixer.in_proj.weight: copying a param with shape torch.Size([1536, 384]) from checkpoint, the shape in current model is torch.Size([3072, 768]).
        size mismatch for layers.15.mixer.conv1d.weight: copying a param with shape torch.Size([768, 1, 4]) from checkpoint, the shape in current model is torch.Size([1536, 1, 4]).
        size mismatch for layers.15.mixer.conv1d.bias: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.15.mixer.x_proj.weight: copying a param with shape torch.Size([56, 768]) from checkpoint, the shape in current model is torch.Size([80, 1536]).
        size mismatch for layers.15.mixer.dt_proj.weight: copying a param with shape torch.Size([768, 24]) from checkpoint, the shape in current model is torch.Size([1536, 48]).
        size mismatch for layers.15.mixer.dt_proj.bias: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.15.mixer.out_proj.weight: copying a param with shape torch.Size([384, 768]) from checkpoint, the shape in current model is torch.Size([768, 1536]).
        size mismatch for layers.15.norm.weight: copying a param with shape torch.Size([384]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for layers.16.mixer.A_log: copying a param with shape torch.Size([768, 16]) from checkpoint, the shape in current model is torch.Size([1536, 16]).
        size mismatch for layers.16.mixer.D: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.16.mixer.in_proj.weight: copying a param with shape torch.Size([1536, 384]) from checkpoint, the shape in current model is torch.Size([3072, 768]).
        size mismatch for layers.16.mixer.conv1d.weight: copying a param with shape torch.Size([768, 1, 4]) from checkpoint, the shape in current model is torch.Size([1536, 1, 4]).
        size mismatch for layers.16.mixer.conv1d.bias: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.16.mixer.x_proj.weight: copying a param with shape torch.Size([56, 768]) from checkpoint, the shape in current model is torch.Size([80, 1536]).
        size mismatch for layers.16.mixer.dt_proj.weight: copying a param with shape torch.Size([768, 24]) from checkpoint, the shape in current model is torch.Size([1536, 48]).
        size mismatch for layers.16.mixer.dt_proj.bias: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.16.mixer.out_proj.weight: copying a param with shape torch.Size([384, 768]) from checkpoint, the shape in current model is torch.Size([768, 1536]).
        size mismatch for layers.16.norm.weight: copying a param with shape torch.Size([384]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for layers.17.mixer.A_log: copying a param with shape torch.Size([768, 16]) from checkpoint, the shape in current model is torch.Size([1536, 16]).
        size mismatch for layers.17.mixer.D: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.17.mixer.in_proj.weight: copying a param with shape torch.Size([1536, 384]) from checkpoint, the shape in current model is torch.Size([3072, 768]).
        size mismatch for layers.17.mixer.conv1d.weight: copying a param with shape torch.Size([768, 1, 4]) from checkpoint, the shape in current model is torch.Size([1536, 1, 4]).
        size mismatch for layers.17.mixer.conv1d.bias: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.17.mixer.x_proj.weight: copying a param with shape torch.Size([56, 768]) from checkpoint, the shape in current model is torch.Size([80, 1536]).
        size mismatch for layers.17.mixer.dt_proj.weight: copying a param with shape torch.Size([768, 24]) from checkpoint, the shape in current model is torch.Size([1536, 48]).
        size mismatch for layers.17.mixer.dt_proj.bias: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.17.mixer.out_proj.weight: copying a param with shape torch.Size([384, 768]) from checkpoint, the shape in current model is torch.Size([768, 1536]).
        size mismatch for layers.17.norm.weight: copying a param with shape torch.Size([384]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for layers.18.mixer.A_log: copying a param with shape torch.Size([768, 16]) from checkpoint, the shape in current model is torch.Size([1536, 16]).
        size mismatch for layers.18.mixer.D: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.18.mixer.in_proj.weight: copying a param with shape torch.Size([1536, 384]) from checkpoint, the shape in current model is torch.Size([3072, 768]).
        size mismatch for layers.18.mixer.conv1d.weight: copying a param with shape torch.Size([768, 1, 4]) from checkpoint, the shape in current model is torch.Size([1536, 1, 4]).
        size mismatch for layers.18.mixer.conv1d.bias: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.18.mixer.x_proj.weight: copying a param with shape torch.Size([56, 768]) from checkpoint, the shape in current model is torch.Size([80, 1536]).
        size mismatch for layers.18.mixer.dt_proj.weight: copying a param with shape torch.Size([768, 24]) from checkpoint, the shape in current model is torch.Size([1536, 48]).
        size mismatch for layers.18.mixer.dt_proj.bias: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.18.mixer.out_proj.weight: copying a param with shape torch.Size([384, 768]) from checkpoint, the shape in current model is torch.Size([768, 1536]).
        size mismatch for layers.18.norm.weight: copying a param with shape torch.Size([384]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for layers.19.mixer.A_log: copying a param with shape torch.Size([768, 16]) from checkpoint, the shape in current model is torch.Size([1536, 16]).
        size mismatch for layers.19.mixer.D: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.19.mixer.in_proj.weight: copying a param with shape torch.Size([1536, 384]) from checkpoint, the shape in current model is torch.Size([3072, 768]).
        size mismatch for layers.19.mixer.conv1d.weight: copying a param with shape torch.Size([768, 1, 4]) from checkpoint, the shape in current model is torch.Size([1536, 1, 4]).
        size mismatch for layers.19.mixer.conv1d.bias: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.19.mixer.x_proj.weight: copying a param with shape torch.Size([56, 768]) from checkpoint, the shape in current model is torch.Size([80, 1536]).
        size mismatch for layers.19.mixer.dt_proj.weight: copying a param with shape torch.Size([768, 24]) from checkpoint, the shape in current model is torch.Size([1536, 48]).
        size mismatch for layers.19.mixer.dt_proj.bias: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.19.mixer.out_proj.weight: copying a param with shape torch.Size([384, 768]) from checkpoint, the shape in current model is torch.Size([768, 1536]).
        size mismatch for layers.19.norm.weight: copying a param with shape torch.Size([384]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for layers.20.mixer.A_log: copying a param with shape torch.Size([768, 16]) from checkpoint, the shape in current model is torch.Size([1536, 16]).
        size mismatch for layers.20.mixer.D: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.20.mixer.in_proj.weight: copying a param with shape torch.Size([1536, 384]) from checkpoint, the shape in current model is torch.Size([3072, 768]).
        size mismatch for layers.20.mixer.conv1d.weight: copying a param with shape torch.Size([768, 1, 4]) from checkpoint, the shape in current model is torch.Size([1536, 1, 4]).
        size mismatch for layers.20.mixer.conv1d.bias: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.20.mixer.x_proj.weight: copying a param with shape torch.Size([56, 768]) from checkpoint, the shape in current model is torch.Size([80, 1536]).
        size mismatch for layers.20.mixer.dt_proj.weight: copying a param with shape torch.Size([768, 24]) from checkpoint, the shape in current model is torch.Size([1536, 48]).
        size mismatch for layers.20.mixer.dt_proj.bias: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.20.mixer.out_proj.weight: copying a param with shape torch.Size([384, 768]) from checkpoint, the shape in current model is torch.Size([768, 1536]).
        size mismatch for layers.20.norm.weight: copying a param with shape torch.Size([384]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for layers.21.mixer.A_log: copying a param with shape torch.Size([768, 16]) from checkpoint, the shape in current model is torch.Size([1536, 16]).
        size mismatch for layers.21.mixer.D: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.21.mixer.in_proj.weight: copying a param with shape torch.Size([1536, 384]) from checkpoint, the shape in current model is torch.Size([3072, 768]).
        size mismatch for layers.21.mixer.conv1d.weight: copying a param with shape torch.Size([768, 1, 4]) from checkpoint, the shape in current model is torch.Size([1536, 1, 4]).
        size mismatch for layers.21.mixer.conv1d.bias: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.21.mixer.x_proj.weight: copying a param with shape torch.Size([56, 768]) from checkpoint, the shape in current model is torch.Size([80, 1536]).
        size mismatch for layers.21.mixer.dt_proj.weight: copying a param with shape torch.Size([768, 24]) from checkpoint, the shape in current model is torch.Size([1536, 48]).
        size mismatch for layers.21.mixer.dt_proj.bias: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.21.mixer.out_proj.weight: copying a param with shape torch.Size([384, 768]) from checkpoint, the shape in current model is torch.Size([768, 1536]).
        size mismatch for layers.21.norm.weight: copying a param with shape torch.Size([384]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for layers.22.mixer.A_log: copying a param with shape torch.Size([768, 16]) from checkpoint, the shape in current model is torch.Size([1536, 16]).
        size mismatch for layers.22.mixer.D: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.22.mixer.in_proj.weight: copying a param with shape torch.Size([1536, 384]) from checkpoint, the shape in current model is torch.Size([3072, 768]).
        size mismatch for layers.22.mixer.conv1d.weight: copying a param with shape torch.Size([768, 1, 4]) from checkpoint, the shape in current model is torch.Size([1536, 1, 4]).
        size mismatch for layers.22.mixer.conv1d.bias: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.22.mixer.x_proj.weight: copying a param with shape torch.Size([56, 768]) from checkpoint, the shape in current model is torch.Size([80, 1536]).
        size mismatch for layers.22.mixer.dt_proj.weight: copying a param with shape torch.Size([768, 24]) from checkpoint, the shape in current model is torch.Size([1536, 48]).
        size mismatch for layers.22.mixer.dt_proj.bias: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.22.mixer.out_proj.weight: copying a param with shape torch.Size([384, 768]) from checkpoint, the shape in current model is torch.Size([768, 1536]).
        size mismatch for layers.22.norm.weight: copying a param with shape torch.Size([384]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for layers.23.mixer.A_log: copying a param with shape torch.Size([768, 16]) from checkpoint, the shape in current model is torch.Size([1536, 16]).
        size mismatch for layers.23.mixer.D: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.23.mixer.in_proj.weight: copying a param with shape torch.Size([1536, 384]) from checkpoint, the shape in current model is torch.Size([3072, 768]).
        size mismatch for layers.23.mixer.conv1d.weight: copying a param with shape torch.Size([768, 1, 4]) from checkpoint, the shape in current model is torch.Size([1536, 1, 4]).
        size mismatch for layers.23.mixer.conv1d.bias: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.23.mixer.x_proj.weight: copying a param with shape torch.Size([56, 768]) from checkpoint, the shape in current model is torch.Size([80, 1536]).
        size mismatch for layers.23.mixer.dt_proj.weight: copying a param with shape torch.Size([768, 24]) from checkpoint, the shape in current model is torch.Size([1536, 48]).
        size mismatch for layers.23.mixer.dt_proj.bias: copying a param with shape torch.Size([768]) from checkpoint, the shape in current model is torch.Size([1536]).
        size mismatch for layers.23.mixer.out_proj.weight: copying a param with shape torch.Size([384, 768]) from checkpoint, the shape in current model is torch.Size([768, 1536]).
        size mismatch for layers.23.norm.weight: copying a param with shape torch.Size([384]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for norm_f.weight: copying a param with shape torch.Size([384]) from checkpoint, the shape in current model is torch.Size([768]).