rs % python submit_multilabel_ensemble.py configs/inference/1128_b5_b4_vs2_fliplr4.yaml
Available keys in model config: dict_keys(['slug', 'activation', 'num_channels', 'num_extra_blocks', 'encoder', 'decoder', 'head', 'box_coder'])
Complete model configuration:
 slug: b4_unet_s2
activation: silu
num_channels: 2
num_extra_blocks: 0
encoder:
  _target_: pytorch_toolbelt.modules.encoders.TimmB4Encoder
  pretrained: true
  activation: silu
  layers:
  - 1
  - 2
  - 3
  - 4
decoder:
  upsample_type: shuffle
  block_type: IRBlock
  channels:
  - 256
  - 384
  - 512
head:
  _target_: xview3.centernet.models.heads.DecoupledHeadGroupNormLateShuffle
  classifier_dim: 128
  objectness_dim: 128
  size_dim: 64
  offset_dim: 16
  dropout_rate: 0.1
  activation: silu
  num_blocks: 3
box_coder:
  _target_: xview3.centernet.MultilabelCircleNetCoder
  image_size:
  - 1024
  - 1024
  max_objects: 512
  heatmap_encoding: umich
  labels_encoding: circle
  ignore_value:
    _target_: xview3.ignore_value
  fixed_radius: 3
  labels_radius: 1
  ignore_low_confidence_detections: true

[DEBUG] Number of heads instantiated: 1
[DEBUG] Head 0 type: <class 'xview3.centernet.models.heads.decoupled_head.DecoupledHeadGroupNormLateShuffle'>
/opt/anaconda3/envs/sam3/lib/python3.9/site-packages/pytorch_toolbelt/modules/encoders/timm/efficient_net.py:181: UserWarning:

Mapping deprecated model name tf_efficientnet_b4_ns to current tf_efficientnet_b4.ns_jft_in1k.

Unexpected keys (bn2.bias, bn2.num_batches_tracked, bn2.running_mean, bn2.running_var, bn2.weight, classifier.bias, classifier.weight, conv_head.weight) found while loading pretrained weights. This may be expected if model is being adapted.
Unexpected keys (bn2.bias, bn2.num_batches_tracked, bn2.running_mean, bn2.running_var, bn2.weight, classifier.bias, classifier.weight, conv_head.weight) found while loading pretrained weights. This may be expected if model is being adapted.
[DEBUG] After adjustment, encoder.conv_stem.in_channels: 2
Total Input Channels for Fusion: 29
head_output_channels: 6
yolo_output_channels: 21
classifier_output_channels: 2
Total Input Channels for Fusion: 29
head_output_channels: 6
yolo_output_channels: 21
classifier_output_channels: 2
Traceback (most recent call last):
  File "/Users/samerkharboush/Downloads/SAR-MPS-OptunaNasFbnYoloTransformers/submit_multilabel_ensemble.py", line 193, in <module>
    Fire(main)
  File "/opt/anaconda3/envs/sam3/lib/python3.9/site-packages/fire/core.py", line 143, in Fire
    component_trace = _Fire(component, args, parsed_flag_args, context, name)
  File "/opt/anaconda3/envs/sam3/lib/python3.9/site-packages/fire/core.py", line 477, in _Fire
    component, remaining_args = _CallAndUpdateTrace(
  File "/opt/anaconda3/envs/sam3/lib/python3.9/site-packages/fire/core.py", line 693, in _CallAndUpdateTrace
    component = fn(*varargs, **kwargs)
  File "/Users/samerkharboush/Downloads/SAR-MPS-OptunaNasFbnYoloTransformers/submit_multilabel_ensemble.py", line 182, in main
    run_multilabel_predict(OmegaConf.load(config), data_dir=data_dir)
  File "/Users/samerkharboush/Downloads/SAR-MPS-OptunaNasFbnYoloTransformers/submit_multilabel_ensemble.py", line 31, in run_multilabel_predict
    model, checkpoints, box_coder = ensemble_from_config(config, map_location=device)
  File "/Users/samerkharboush/Downloads/SAR-MPS-OptunaNasFbnYoloTransformers/xview3/inference.py", line 241, in ensemble_from_config
    model, checkpoints = ensemble_from_checkpoints(
  File "/Users/samerkharboush/Downloads/SAR-MPS-OptunaNasFbnYoloTransformers/xview3/inference.py", line 198, in ensemble_from_checkpoints
    model, checkpoint = model_from_checkpoint(ck, strict=strict, map_location=map_location)
  File "/Users/samerkharboush/Downloads/SAR-MPS-OptunaNasFbnYoloTransformers/xview3/inference.py", line 135, in model_from_checkpoint
    model.load_state_dict(model_state_dict, strict=False)
  File "/opt/anaconda3/envs/sam3/lib/python3.9/site-packages/torch/nn/modules/module.py", line 2215, in load_state_dict
    raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format(
RuntimeError: Error(s) in loading state_dict for CenterNetUNetModel:
        size mismatch for encoder.encoder.blocks.1.0.conv_pwl.weight: copying a param with shape torch.Size([32, 144, 1, 1]) from checkpoint, the shape in current model is torch.Size([40, 144, 1, 1]).
        size mismatch for encoder.encoder.blocks.1.0.bn3.weight: copying a param with shape torch.Size([32]) from checkpoint, the shape in current model is torch.Size([40]).
        size mismatch for encoder.encoder.blocks.1.0.bn3.bias: copying a param with shape torch.Size([32]) from checkpoint, the shape in current model is torch.Size([40]).
        size mismatch for encoder.encoder.blocks.1.0.bn3.running_mean: copying a param with shape torch.Size([32]) from checkpoint, the shape in current model is torch.Size([40]).
        size mismatch for encoder.encoder.blocks.1.0.bn3.running_var: copying a param with shape torch.Size([32]) from checkpoint, the shape in current model is torch.Size([40]).
        size mismatch for encoder.encoder.blocks.1.1.conv_pw.weight: copying a param with shape torch.Size([192, 32, 1, 1]) from checkpoint, the shape in current model is torch.Size([240, 40, 1, 1]).
        size mismatch for encoder.encoder.blocks.1.1.bn1.weight: copying a param with shape torch.Size([192]) from checkpoint, the shape in current model is torch.Size([240]).
        size mismatch for encoder.encoder.blocks.1.1.bn1.bias: copying a param with shape torch.Size([192]) from checkpoint, the shape in current model is torch.Size([240]).
        size mismatch for encoder.encoder.blocks.1.1.bn1.running_mean: copying a param with shape torch.Size([192]) from checkpoint, the shape in current model is torch.Size([240]).
        size mismatch for encoder.encoder.blocks.1.1.bn1.running_var: copying a param with shape torch.Size([192]) from checkpoint, the shape in current model is torch.Size([240]).
        size mismatch for encoder.encoder.blocks.1.1.conv_dw.weight: copying a param with shape torch.Size([192, 1, 3, 3]) from checkpoint, the shape in current model is torch.Size([240, 1, 3, 3]).
        size mismatch for encoder.encoder.blocks.1.1.bn2.weight: copying a param with shape torch.Size([192]) from checkpoint, the shape in current model is torch.Size([240]).
        size mismatch for encoder.encoder.blocks.1.1.bn2.bias: copying a param with shape torch.Size([192]) from checkpoint, the shape in current model is torch.Size([240]).
        size mismatch for encoder.encoder.blocks.1.1.bn2.running_mean: copying a param with shape torch.Size([192]) from checkpoint, the shape in current model is torch.Size([240]).
        size mismatch for encoder.encoder.blocks.1.1.bn2.running_var: copying a param with shape torch.Size([192]) from checkpoint, the shape in current model is torch.Size([240]).
        size mismatch for encoder.encoder.blocks.1.1.se.conv_reduce.weight: copying a param with shape torch.Size([8, 192, 1, 1]) from checkpoint, the shape in current model is torch.Size([10, 240, 1, 1]).
        size mismatch for encoder.encoder.blocks.1.1.se.conv_reduce.bias: copying a param with shape torch.Size([8]) from checkpoint, the shape in current model is torch.Size([10]).
        size mismatch for encoder.encoder.blocks.1.1.se.conv_expand.weight: copying a param with shape torch.Size([192, 8, 1, 1]) from checkpoint, the shape in current model is torch.Size([240, 10, 1, 1]).
        size mismatch for encoder.encoder.blocks.1.1.se.conv_expand.bias: copying a param with shape torch.Size([192]) from checkpoint, the shape in current model is torch.Size([240]).
        size mismatch for encoder.encoder.blocks.1.1.conv_pwl.weight: copying a param with shape torch.Size([32, 192, 1, 1]) from checkpoint, the shape in current model is torch.Size([40, 240, 1, 1]).
        size mismatch for encoder.encoder.blocks.1.1.bn3.weight: copying a param with shape torch.Size([32]) from checkpoint, the shape in current model is torch.Size([40]).
        size mismatch for encoder.encoder.blocks.1.1.bn3.bias: copying a param with shape torch.Size([32]) from checkpoint, the shape in current model is torch.Size([40]).
        size mismatch for encoder.encoder.blocks.1.1.bn3.running_mean: copying a param with shape torch.Size([32]) from checkpoint, the shape in current model is torch.Size([40]).
        size mismatch for encoder.encoder.blocks.1.1.bn3.running_var: copying a param with shape torch.Size([32]) from checkpoint, the shape in current model is torch.Size([40]).
        size mismatch for encoder.encoder.blocks.1.2.conv_pw.weight: copying a param with shape torch.Size([192, 32, 1, 1]) from checkpoint, the shape in current model is torch.Size([240, 40, 1, 1]).
        size mismatch for encoder.encoder.blocks.1.2.bn1.weight: copying a param with shape torch.Size([192]) from checkpoint, the shape in current model is torch.Size([240]).
        size mismatch for encoder.encoder.blocks.1.2.bn1.bias: copying a param with shape torch.Size([192]) from checkpoint, the shape in current model is torch.Size([240]).
        size mismatch for encoder.encoder.blocks.1.2.bn1.running_mean: copying a param with shape torch.Size([192]) from checkpoint, the shape in current model is torch.Size([240]).
        size mismatch for encoder.encoder.blocks.1.2.bn1.running_var: copying a param with shape torch.Size([192]) from checkpoint, the shape in current model is torch.Size([240]).
        size mismatch for encoder.encoder.blocks.1.2.conv_dw.weight: copying a param with shape torch.Size([192, 1, 3, 3]) from checkpoint, the shape in current model is torch.Size([240, 1, 3, 3]).
        size mismatch for encoder.encoder.blocks.1.2.bn2.weight: copying a param with shape torch.Size([192]) from checkpoint, the shape in current model is torch.Size([240]).
        size mismatch for encoder.encoder.blocks.1.2.bn2.bias: copying a param with shape torch.Size([192]) from checkpoint, the shape in current model is torch.Size([240]).
        size mismatch for encoder.encoder.blocks.1.2.bn2.running_mean: copying a param with shape torch.Size([192]) from checkpoint, the shape in current model is torch.Size([240]).
        size mismatch for encoder.encoder.blocks.1.2.bn2.running_var: copying a param with shape torch.Size([192]) from checkpoint, the shape in current model is torch.Size([240]).
        size mismatch for encoder.encoder.blocks.1.2.se.conv_reduce.weight: copying a param with shape torch.Size([8, 192, 1, 1]) from checkpoint, the shape in current model is torch.Size([10, 240, 1, 1]).
        size mismatch for encoder.encoder.blocks.1.2.se.conv_reduce.bias: copying a param with shape torch.Size([8]) from checkpoint, the shape in current model is torch.Size([10]).
        size mismatch for encoder.encoder.blocks.1.2.se.conv_expand.weight: copying a param with shape torch.Size([192, 8, 1, 1]) from checkpoint, the shape in current model is torch.Size([240, 10, 1, 1]).
        size mismatch for encoder.encoder.blocks.1.2.se.conv_expand.bias: copying a param with shape torch.Size([192]) from checkpoint, the shape in current model is torch.Size([240]).
        size mismatch for encoder.encoder.blocks.1.2.conv_pwl.weight: copying a param with shape torch.Size([32, 192, 1, 1]) from checkpoint, the shape in current model is torch.Size([40, 240, 1, 1]).
        size mismatch for encoder.encoder.blocks.1.2.bn3.weight: copying a param with shape torch.Size([32]) from checkpoint, the shape in current model is torch.Size([40]).
        size mismatch for encoder.encoder.blocks.1.2.bn3.bias: copying a param with shape torch.Size([32]) from checkpoint, the shape in current model is torch.Size([40]).
        size mismatch for encoder.encoder.blocks.1.2.bn3.running_mean: copying a param with shape torch.Size([32]) from checkpoint, the shape in current model is torch.Size([40]).
        size mismatch for encoder.encoder.blocks.1.2.bn3.running_var: copying a param with shape torch.Size([32]) from checkpoint, the shape in current model is torch.Size([40]).
        size mismatch for encoder.encoder.blocks.1.3.conv_pw.weight: copying a param with shape torch.Size([192, 32, 1, 1]) from checkpoint, the shape in current model is torch.Size([240, 40, 1, 1]).
        size mismatch for encoder.encoder.blocks.1.3.bn1.weight: copying a param with shape torch.Size([192]) from checkpoint, the shape in current model is torch.Size([240]).
        size mismatch for encoder.encoder.blocks.1.3.bn1.bias: copying a param with shape torch.Size([192]) from checkpoint, the shape in current model is torch.Size([240]).
        size mismatch for encoder.encoder.blocks.1.3.bn1.running_mean: copying a param with shape torch.Size([192]) from checkpoint, the shape in current model is torch.Size([240]).
        size mismatch for encoder.encoder.blocks.1.3.bn1.running_var: copying a param with shape torch.Size([192]) from checkpoint, the shape in current model is torch.Size([240]).
        size mismatch for encoder.encoder.blocks.1.3.conv_dw.weight: copying a param with shape torch.Size([192, 1, 3, 3]) from checkpoint, the shape in current model is torch.Size([240, 1, 3, 3]).
        size mismatch for encoder.encoder.blocks.1.3.bn2.weight: copying a param with shape torch.Size([192]) from checkpoint, the shape in current model is torch.Size([240]).
        size mismatch for encoder.encoder.blocks.1.3.bn2.bias: copying a param with shape torch.Size([192]) from checkpoint, the shape in current model is torch.Size([240]).
        size mismatch for encoder.encoder.blocks.1.3.bn2.running_mean: copying a param with shape torch.Size([192]) from checkpoint, the shape in current model is torch.Size([240]).
        size mismatch for encoder.encoder.blocks.1.3.bn2.running_var: copying a param with shape torch.Size([192]) from checkpoint, the shape in current model is torch.Size([240]).
        size mismatch for encoder.encoder.blocks.1.3.se.conv_reduce.weight: copying a param with shape torch.Size([8, 192, 1, 1]) from checkpoint, the shape in current model is torch.Size([10, 240, 1, 1]).
        size mismatch for encoder.encoder.blocks.1.3.se.conv_reduce.bias: copying a param with shape torch.Size([8]) from checkpoint, the shape in current model is torch.Size([10]).
        size mismatch for encoder.encoder.blocks.1.3.se.conv_expand.weight: copying a param with shape torch.Size([192, 8, 1, 1]) from checkpoint, the shape in current model is torch.Size([240, 10, 1, 1]).
        size mismatch for encoder.encoder.blocks.1.3.se.conv_expand.bias: copying a param with shape torch.Size([192]) from checkpoint, the shape in current model is torch.Size([240]).
        size mismatch for encoder.encoder.blocks.1.3.conv_pwl.weight: copying a param with shape torch.Size([32, 192, 1, 1]) from checkpoint, the shape in current model is torch.Size([40, 240, 1, 1]).
        size mismatch for encoder.encoder.blocks.1.3.bn3.weight: copying a param with shape torch.Size([32]) from checkpoint, the shape in current model is torch.Size([40]).
        size mismatch for encoder.encoder.blocks.1.3.bn3.bias: copying a param with shape torch.Size([32]) from checkpoint, the shape in current model is torch.Size([40]).
        size mismatch for encoder.encoder.blocks.1.3.bn3.running_mean: copying a param with shape torch.Size([32]) from checkpoint, the shape in current model is torch.Size([40]).
        size mismatch for encoder.encoder.blocks.1.3.bn3.running_var: copying a param with shape torch.Size([32]) from checkpoint, the shape in current model is torch.Size([40]).
        size mismatch for encoder.encoder.blocks.2.0.conv_pw.weight: copying a param with shape torch.Size([192, 32, 1, 1]) from checkpoint, the shape in current model is torch.Size([240, 40, 1, 1]).
        size mismatch for encoder.encoder.blocks.2.0.bn1.weight: copying a param with shape torch.Size([192]) from checkpoint, the shape in current model is torch.Size([240]).
        size mismatch for encoder.encoder.blocks.2.0.bn1.bias: copying a param with shape torch.Size([192]) from checkpoint, the shape in current model is torch.Size([240]).
        size mismatch for encoder.encoder.blocks.2.0.bn1.running_mean: copying a param with shape torch.Size([192]) from checkpoint, the shape in current model is torch.Size([240]).
        size mismatch for encoder.encoder.blocks.2.0.bn1.running_var: copying a param with shape torch.Size([192]) from checkpoint, the shape in current model is torch.Size([240]).
        size mismatch for encoder.encoder.blocks.2.0.conv_dw.weight: copying a param with shape torch.Size([192, 1, 5, 5]) from checkpoint, the shape in current model is torch.Size([240, 1, 5, 5]).
        size mismatch for encoder.encoder.blocks.2.0.bn2.weight: copying a param with shape torch.Size([192]) from checkpoint, the shape in current model is torch.Size([240]).
        size mismatch for encoder.encoder.blocks.2.0.bn2.bias: copying a param with shape torch.Size([192]) from checkpoint, the shape in current model is torch.Size([240]).
        size mismatch for encoder.encoder.blocks.2.0.bn2.running_mean: copying a param with shape torch.Size([192]) from checkpoint, the shape in current model is torch.Size([240]).
        size mismatch for encoder.encoder.blocks.2.0.bn2.running_var: copying a param with shape torch.Size([192]) from checkpoint, the shape in current model is torch.Size([240]).
        size mismatch for encoder.encoder.blocks.2.0.se.conv_reduce.weight: copying a param with shape torch.Size([8, 192, 1, 1]) from checkpoint, the shape in current model is torch.Size([10, 240, 1, 1]).
        size mismatch for encoder.encoder.blocks.2.0.se.conv_reduce.bias: copying a param with shape torch.Size([8]) from checkpoint, the shape in current model is torch.Size([10]).
        size mismatch for encoder.encoder.blocks.2.0.se.conv_expand.weight: copying a param with shape torch.Size([192, 8, 1, 1]) from checkpoint, the shape in current model is torch.Size([240, 10, 1, 1]).
        size mismatch for encoder.encoder.blocks.2.0.se.conv_expand.bias: copying a param with shape torch.Size([192]) from checkpoint, the shape in current model is torch.Size([240]).
        size mismatch for encoder.encoder.blocks.2.0.conv_pwl.weight: copying a param with shape torch.Size([56, 192, 1, 1]) from checkpoint, the shape in current model is torch.Size([64, 240, 1, 1]).
        size mismatch for encoder.encoder.blocks.2.0.bn3.weight: copying a param with shape torch.Size([56]) from checkpoint, the shape in current model is torch.Size([64]).
        size mismatch for encoder.encoder.blocks.2.0.bn3.bias: copying a param with shape torch.Size([56]) from checkpoint, the shape in current model is torch.Size([64]).
        size mismatch for encoder.encoder.blocks.2.0.bn3.running_mean: copying a param with shape torch.Size([56]) from checkpoint, the shape in current model is torch.Size([64]).
        size mismatch for encoder.encoder.blocks.2.0.bn3.running_var: copying a param with shape torch.Size([56]) from checkpoint, the shape in current model is torch.Size([64]).
        size mismatch for encoder.encoder.blocks.2.1.conv_pw.weight: copying a param with shape torch.Size([336, 56, 1, 1]) from checkpoint, the shape in current model is torch.Size([384, 64, 1, 1]).
        size mismatch for encoder.encoder.blocks.2.1.bn1.weight: copying a param with shape torch.Size([336]) from checkpoint, the shape in current model is torch.Size([384]).
        size mismatch for encoder.encoder.blocks.2.1.bn1.bias: copying a param with shape torch.Size([336]) from checkpoint, the shape in current model is torch.Size([384]).
        size mismatch for encoder.encoder.blocks.2.1.bn1.running_mean: copying a param with shape torch.Size([336]) from checkpoint, the shape in current model is torch.Size([384]).
        size mismatch for encoder.encoder.blocks.2.1.bn1.running_var: copying a param with shape torch.Size([336]) from checkpoint, the shape in current model is torch.Size([384]).
        size mismatch for encoder.encoder.blocks.2.1.conv_dw.weight: copying a param with shape torch.Size([336, 1, 5, 5]) from checkpoint, the shape in current model is torch.Size([384, 1, 5, 5]).
        size mismatch for encoder.encoder.blocks.2.1.bn2.weight: copying a param with shape torch.Size([336]) from checkpoint, the shape in current model is torch.Size([384]).
        size mismatch for encoder.encoder.blocks.2.1.bn2.bias: copying a param with shape torch.Size([336]) from checkpoint, the shape in current model is torch.Size([384]).
        size mismatch for encoder.encoder.blocks.2.1.bn2.running_mean: copying a param with shape torch.Size([336]) from checkpoint, the shape in current model is torch.Size([384]).
        size mismatch for encoder.encoder.blocks.2.1.bn2.running_var: copying a param with shape torch.Size([336]) from checkpoint, the shape in current model is torch.Size([384]).
        size mismatch for encoder.encoder.blocks.2.1.se.conv_reduce.weight: copying a param with shape torch.Size([14, 336, 1, 1]) from checkpoint, the shape in current model is torch.Size([16, 384, 1, 1]).
        size mismatch for encoder.encoder.blocks.2.1.se.conv_reduce.bias: copying a param with shape torch.Size([14]) from checkpoint, the shape in current model is torch.Size([16]).
        size mismatch for encoder.encoder.blocks.2.1.se.conv_expand.weight: copying a param with shape torch.Size([336, 14, 1, 1]) from checkpoint, the shape in current model is torch.Size([384, 16, 1, 1]).
        size mismatch for encoder.encoder.blocks.2.1.se.conv_expand.bias: copying a param with shape torch.Size([336]) from checkpoint, the shape in current model is torch.Size([384]).
        size mismatch for encoder.encoder.blocks.2.1.conv_pwl.weight: copying a param with shape torch.Size([56, 336, 1, 1]) from checkpoint, the shape in current model is torch.Size([64, 384, 1, 1]).
        size mismatch for encoder.encoder.blocks.2.1.bn3.weight: copying a param with shape torch.Size([56]) from checkpoint, the shape in current model is torch.Size([64]).
        size mismatch for encoder.encoder.blocks.2.1.bn3.bias: copying a param with shape torch.Size([56]) from checkpoint, the shape in current model is torch.Size([64]).
        size mismatch for encoder.encoder.blocks.2.1.bn3.running_mean: copying a param with shape torch.Size([56]) from checkpoint, the shape in current model is torch.Size([64]).
        size mismatch for encoder.encoder.blocks.2.1.bn3.running_var: copying a param with shape torch.Size([56]) from checkpoint, the shape in current model is torch.Size([64]).
        size mismatch for encoder.encoder.blocks.2.2.conv_pw.weight: copying a param with shape torch.Size([336, 56, 1, 1]) from checkpoint, the shape in current model is torch.Size([384, 64, 1, 1]).
        size mismatch for encoder.encoder.blocks.2.2.bn1.weight: copying a param with shape torch.Size([336]) from checkpoint, the shape in current model is torch.Size([384]).
        size mismatch for encoder.encoder.blocks.2.2.bn1.bias: copying a param with shape torch.Size([336]) from checkpoint, the shape in current model is torch.Size([384]).
        size mismatch for encoder.encoder.blocks.2.2.bn1.running_mean: copying a param with shape torch.Size([336]) from checkpoint, the shape in current model is torch.Size([384]).
        size mismatch for encoder.encoder.blocks.2.2.bn1.running_var: copying a param with shape torch.Size([336]) from checkpoint, the shape in current model is torch.Size([384]).
        size mismatch for encoder.encoder.blocks.2.2.conv_dw.weight: copying a param with shape torch.Size([336, 1, 5, 5]) from checkpoint, the shape in current model is torch.Size([384, 1, 5, 5]).
        size mismatch for encoder.encoder.blocks.2.2.bn2.weight: copying a param with shape torch.Size([336]) from checkpoint, the shape in current model is torch.Size([384]).
        size mismatch for encoder.encoder.blocks.2.2.bn2.bias: copying a param with shape torch.Size([336]) from checkpoint, the shape in current model is torch.Size([384]).
        size mismatch for encoder.encoder.blocks.2.2.bn2.running_mean: copying a param with shape torch.Size([336]) from checkpoint, the shape in current model is torch.Size([384]).
        size mismatch for encoder.encoder.blocks.2.2.bn2.running_var: copying a param with shape torch.Size([336]) from checkpoint, the shape in current model is torch.Size([384]).
        size mismatch for encoder.encoder.blocks.2.2.se.conv_reduce.weight: copying a param with shape torch.Size([14, 336, 1, 1]) from checkpoint, the shape in current model is torch.Size([16, 384, 1, 1]).
        size mismatch for encoder.encoder.blocks.2.2.se.conv_reduce.bias: copying a param with shape torch.Size([14]) from checkpoint, the shape in current model is torch.Size([16]).
        size mismatch for encoder.encoder.blocks.2.2.se.conv_expand.weight: copying a param with shape torch.Size([336, 14, 1, 1]) from checkpoint, the shape in current model is torch.Size([384, 16, 1, 1]).
        size mismatch for encoder.encoder.blocks.2.2.se.conv_expand.bias: copying a param with shape torch.Size([336]) from checkpoint, the shape in current model is torch.Size([384]).
        size mismatch for encoder.encoder.blocks.2.2.conv_pwl.weight: copying a param with shape torch.Size([56, 336, 1, 1]) from checkpoint, the shape in current model is torch.Size([64, 384, 1, 1]).
        size mismatch for encoder.encoder.blocks.2.2.bn3.weight: copying a param with shape torch.Size([56]) from checkpoint, the shape in current model is torch.Size([64]).
        size mismatch for encoder.encoder.blocks.2.2.bn3.bias: copying a param with shape torch.Size([56]) from checkpoint, the shape in current model is torch.Size([64]).
        size mismatch for encoder.encoder.blocks.2.2.bn3.running_mean: copying a param with shape torch.Size([56]) from checkpoint, the shape in current model is torch.Size([64]).
        size mismatch for encoder.encoder.blocks.2.2.bn3.running_var: copying a param with shape torch.Size([56]) from checkpoint, the shape in current model is torch.Size([64]).
        size mismatch for encoder.encoder.blocks.2.3.conv_pw.weight: copying a param with shape torch.Size([336, 56, 1, 1]) from checkpoint, the shape in current model is torch.Size([384, 64, 1, 1]).
        size mismatch for encoder.encoder.blocks.2.3.bn1.weight: copying a param with shape torch.Size([336]) from checkpoint, the shape in current model is torch.Size([384]).
        size mismatch for encoder.encoder.blocks.2.3.bn1.bias: copying a param with shape torch.Size([336]) from checkpoint, the shape in current model is torch.Size([384]).
        size mismatch for encoder.encoder.blocks.2.3.bn1.running_mean: copying a param with shape torch.Size([336]) from checkpoint, the shape in current model is torch.Size([384]).
        size mismatch for encoder.encoder.blocks.2.3.bn1.running_var: copying a param with shape torch.Size([336]) from checkpoint, the shape in current model is torch.Size([384]).
        size mismatch for encoder.encoder.blocks.2.3.conv_dw.weight: copying a param with shape torch.Size([336, 1, 5, 5]) from checkpoint, the shape in current model is torch.Size([384, 1, 5, 5]).
        size mismatch for encoder.encoder.blocks.2.3.bn2.weight: copying a param with shape torch.Size([336]) from checkpoint, the shape in current model is torch.Size([384]).
        size mismatch for encoder.encoder.blocks.2.3.bn2.bias: copying a param with shape torch.Size([336]) from checkpoint, the shape in current model is torch.Size([384]).
        size mismatch for encoder.encoder.blocks.2.3.bn2.running_mean: copying a param with shape torch.Size([336]) from checkpoint, the shape in current model is torch.Size([384]).
        size mismatch for encoder.encoder.blocks.2.3.bn2.running_var: copying a param with shape torch.Size([336]) from checkpoint, the shape in current model is torch.Size([384]).
        size mismatch for encoder.encoder.blocks.2.3.se.conv_reduce.weight: copying a param with shape torch.Size([14, 336, 1, 1]) from checkpoint, the shape in current model is torch.Size([16, 384, 1, 1]).
        size mismatch for encoder.encoder.blocks.2.3.se.conv_reduce.bias: copying a param with shape torch.Size([14]) from checkpoint, the shape in current model is torch.Size([16]).
        size mismatch for encoder.encoder.blocks.2.3.se.conv_expand.weight: copying a param with shape torch.Size([336, 14, 1, 1]) from checkpoint, the shape in current model is torch.Size([384, 16, 1, 1]).
        size mismatch for encoder.encoder.blocks.2.3.se.conv_expand.bias: copying a param with shape torch.Size([336]) from checkpoint, the shape in current model is torch.Size([384]).
        size mismatch for encoder.encoder.blocks.2.3.conv_pwl.weight: copying a param with shape torch.Size([56, 336, 1, 1]) from checkpoint, the shape in current model is torch.Size([64, 384, 1, 1]).
        size mismatch for encoder.encoder.blocks.2.3.bn3.weight: copying a param with shape torch.Size([56]) from checkpoint, the shape in current model is torch.Size([64]).
        size mismatch for encoder.encoder.blocks.2.3.bn3.bias: copying a param with shape torch.Size([56]) from checkpoint, the shape in current model is torch.Size([64]).
        size mismatch for encoder.encoder.blocks.2.3.bn3.running_mean: copying a param with shape torch.Size([56]) from checkpoint, the shape in current model is torch.Size([64]).
        size mismatch for encoder.encoder.blocks.2.3.bn3.running_var: copying a param with shape torch.Size([56]) from checkpoint, the shape in current model is torch.Size([64]).
        size mismatch for encoder.encoder.blocks.3.0.conv_pw.weight: copying a param with shape torch.Size([336, 56, 1, 1]) from checkpoint, the shape in current model is torch.Size([384, 64, 1, 1]).
        size mismatch for encoder.encoder.blocks.3.0.bn1.weight: copying a param with shape torch.Size([336]) from checkpoint, the shape in current model is torch.Size([384]).
        size mismatch for encoder.encoder.blocks.3.0.bn1.bias: copying a param with shape torch.Size([336]) from checkpoint, the shape in current model is torch.Size([384]).
        size mismatch for encoder.encoder.blocks.3.0.bn1.running_mean: copying a param with shape torch.Size([336]) from checkpoint, the shape in current model is torch.Size([384]).
        size mismatch for encoder.encoder.blocks.3.0.bn1.running_var: copying a param with shape torch.Size([336]) from checkpoint, the shape in current model is torch.Size([384]).
        size mismatch for encoder.encoder.blocks.3.0.conv_dw.weight: copying a param with shape torch.Size([336, 1, 3, 3]) from checkpoint, the shape in current model is torch.Size([384, 1, 3, 3]).
        size mismatch for encoder.encoder.blocks.3.0.bn2.weight: copying a param with shape torch.Size([336]) from checkpoint, the shape in current model is torch.Size([384]).
        size mismatch for encoder.encoder.blocks.3.0.bn2.bias: copying a param with shape torch.Size([336]) from checkpoint, the shape in current model is torch.Size([384]).
        size mismatch for encoder.encoder.blocks.3.0.bn2.running_mean: copying a param with shape torch.Size([336]) from checkpoint, the shape in current model is torch.Size([384]).
        size mismatch for encoder.encoder.blocks.3.0.bn2.running_var: copying a param with shape torch.Size([336]) from checkpoint, the shape in current model is torch.Size([384]).
        size mismatch for encoder.encoder.blocks.3.0.se.conv_reduce.weight: copying a param with shape torch.Size([14, 336, 1, 1]) from checkpoint, the shape in current model is torch.Size([16, 384, 1, 1]).
        size mismatch for encoder.encoder.blocks.3.0.se.conv_reduce.bias: copying a param with shape torch.Size([14]) from checkpoint, the shape in current model is torch.Size([16]).
        size mismatch for encoder.encoder.blocks.3.0.se.conv_expand.weight: copying a param with shape torch.Size([336, 14, 1, 1]) from checkpoint, the shape in current model is torch.Size([384, 16, 1, 1]).
        size mismatch for encoder.encoder.blocks.3.0.se.conv_expand.bias: copying a param with shape torch.Size([336]) from checkpoint, the shape in current model is torch.Size([384]).
        size mismatch for encoder.encoder.blocks.3.0.conv_pwl.weight: copying a param with shape torch.Size([112, 336, 1, 1]) from checkpoint, the shape in current model is torch.Size([128, 384, 1, 1]).
        size mismatch for encoder.encoder.blocks.3.0.bn3.weight: copying a param with shape torch.Size([112]) from checkpoint, the shape in current model is torch.Size([128]).
        size mismatch for encoder.encoder.blocks.3.0.bn3.bias: copying a param with shape torch.Size([112]) from checkpoint, the shape in current model is torch.Size([128]).
        size mismatch for encoder.encoder.blocks.3.0.bn3.running_mean: copying a param with shape torch.Size([112]) from checkpoint, the shape in current model is torch.Size([128]).
        size mismatch for encoder.encoder.blocks.3.0.bn3.running_var: copying a param with shape torch.Size([112]) from checkpoint, the shape in current model is torch.Size([128]).
        size mismatch for encoder.encoder.blocks.3.1.conv_pw.weight: copying a param with shape torch.Size([672, 112, 1, 1]) from checkpoint, the shape in current model is torch.Size([768, 128, 1, 1]).
        size mismatch for encoder.encoder.blocks.3.1.bn1.weight: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for encoder.encoder.blocks.3.1.bn1.bias: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for encoder.encoder.blocks.3.1.bn1.running_mean: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for encoder.encoder.blocks.3.1.bn1.running_var: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for encoder.encoder.blocks.3.1.conv_dw.weight: copying a param with shape torch.Size([672, 1, 3, 3]) from checkpoint, the shape in current model is torch.Size([768, 1, 3, 3]).
        size mismatch for encoder.encoder.blocks.3.1.bn2.weight: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for encoder.encoder.blocks.3.1.bn2.bias: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for encoder.encoder.blocks.3.1.bn2.running_mean: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for encoder.encoder.blocks.3.1.bn2.running_var: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for encoder.encoder.blocks.3.1.se.conv_reduce.weight: copying a param with shape torch.Size([28, 672, 1, 1]) from checkpoint, the shape in current model is torch.Size([32, 768, 1, 1]).
        size mismatch for encoder.encoder.blocks.3.1.se.conv_reduce.bias: copying a param with shape torch.Size([28]) from checkpoint, the shape in current model is torch.Size([32]).
        size mismatch for encoder.encoder.blocks.3.1.se.conv_expand.weight: copying a param with shape torch.Size([672, 28, 1, 1]) from checkpoint, the shape in current model is torch.Size([768, 32, 1, 1]).
        size mismatch for encoder.encoder.blocks.3.1.se.conv_expand.bias: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for encoder.encoder.blocks.3.1.conv_pwl.weight: copying a param with shape torch.Size([112, 672, 1, 1]) from checkpoint, the shape in current model is torch.Size([128, 768, 1, 1]).
        size mismatch for encoder.encoder.blocks.3.1.bn3.weight: copying a param with shape torch.Size([112]) from checkpoint, the shape in current model is torch.Size([128]).
        size mismatch for encoder.encoder.blocks.3.1.bn3.bias: copying a param with shape torch.Size([112]) from checkpoint, the shape in current model is torch.Size([128]).
        size mismatch for encoder.encoder.blocks.3.1.bn3.running_mean: copying a param with shape torch.Size([112]) from checkpoint, the shape in current model is torch.Size([128]).
        size mismatch for encoder.encoder.blocks.3.1.bn3.running_var: copying a param with shape torch.Size([112]) from checkpoint, the shape in current model is torch.Size([128]).
        size mismatch for encoder.encoder.blocks.3.2.conv_pw.weight: copying a param with shape torch.Size([672, 112, 1, 1]) from checkpoint, the shape in current model is torch.Size([768, 128, 1, 1]).
        size mismatch for encoder.encoder.blocks.3.2.bn1.weight: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for encoder.encoder.blocks.3.2.bn1.bias: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for encoder.encoder.blocks.3.2.bn1.running_mean: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for encoder.encoder.blocks.3.2.bn1.running_var: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for encoder.encoder.blocks.3.2.conv_dw.weight: copying a param with shape torch.Size([672, 1, 3, 3]) from checkpoint, the shape in current model is torch.Size([768, 1, 3, 3]).
        size mismatch for encoder.encoder.blocks.3.2.bn2.weight: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for encoder.encoder.blocks.3.2.bn2.bias: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for encoder.encoder.blocks.3.2.bn2.running_mean: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for encoder.encoder.blocks.3.2.bn2.running_var: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for encoder.encoder.blocks.3.2.se.conv_reduce.weight: copying a param with shape torch.Size([28, 672, 1, 1]) from checkpoint, the shape in current model is torch.Size([32, 768, 1, 1]).
        size mismatch for encoder.encoder.blocks.3.2.se.conv_reduce.bias: copying a param with shape torch.Size([28]) from checkpoint, the shape in current model is torch.Size([32]).
        size mismatch for encoder.encoder.blocks.3.2.se.conv_expand.weight: copying a param with shape torch.Size([672, 28, 1, 1]) from checkpoint, the shape in current model is torch.Size([768, 32, 1, 1]).
        size mismatch for encoder.encoder.blocks.3.2.se.conv_expand.bias: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for encoder.encoder.blocks.3.2.conv_pwl.weight: copying a param with shape torch.Size([112, 672, 1, 1]) from checkpoint, the shape in current model is torch.Size([128, 768, 1, 1]).
        size mismatch for encoder.encoder.blocks.3.2.bn3.weight: copying a param with shape torch.Size([112]) from checkpoint, the shape in current model is torch.Size([128]).
        size mismatch for encoder.encoder.blocks.3.2.bn3.bias: copying a param with shape torch.Size([112]) from checkpoint, the shape in current model is torch.Size([128]).
        size mismatch for encoder.encoder.blocks.3.2.bn3.running_mean: copying a param with shape torch.Size([112]) from checkpoint, the shape in current model is torch.Size([128]).
        size mismatch for encoder.encoder.blocks.3.2.bn3.running_var: copying a param with shape torch.Size([112]) from checkpoint, the shape in current model is torch.Size([128]).
        size mismatch for encoder.encoder.blocks.3.3.conv_pw.weight: copying a param with shape torch.Size([672, 112, 1, 1]) from checkpoint, the shape in current model is torch.Size([768, 128, 1, 1]).
        size mismatch for encoder.encoder.blocks.3.3.bn1.weight: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for encoder.encoder.blocks.3.3.bn1.bias: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for encoder.encoder.blocks.3.3.bn1.running_mean: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for encoder.encoder.blocks.3.3.bn1.running_var: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for encoder.encoder.blocks.3.3.conv_dw.weight: copying a param with shape torch.Size([672, 1, 3, 3]) from checkpoint, the shape in current model is torch.Size([768, 1, 3, 3]).
        size mismatch for encoder.encoder.blocks.3.3.bn2.weight: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for encoder.encoder.blocks.3.3.bn2.bias: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for encoder.encoder.blocks.3.3.bn2.running_mean: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for encoder.encoder.blocks.3.3.bn2.running_var: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for encoder.encoder.blocks.3.3.se.conv_reduce.weight: copying a param with shape torch.Size([28, 672, 1, 1]) from checkpoint, the shape in current model is torch.Size([32, 768, 1, 1]).
        size mismatch for encoder.encoder.blocks.3.3.se.conv_reduce.bias: copying a param with shape torch.Size([28]) from checkpoint, the shape in current model is torch.Size([32]).
        size mismatch for encoder.encoder.blocks.3.3.se.conv_expand.weight: copying a param with shape torch.Size([672, 28, 1, 1]) from checkpoint, the shape in current model is torch.Size([768, 32, 1, 1]).
        size mismatch for encoder.encoder.blocks.3.3.se.conv_expand.bias: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for encoder.encoder.blocks.3.3.conv_pwl.weight: copying a param with shape torch.Size([112, 672, 1, 1]) from checkpoint, the shape in current model is torch.Size([128, 768, 1, 1]).
        size mismatch for encoder.encoder.blocks.3.3.bn3.weight: copying a param with shape torch.Size([112]) from checkpoint, the shape in current model is torch.Size([128]).
        size mismatch for encoder.encoder.blocks.3.3.bn3.bias: copying a param with shape torch.Size([112]) from checkpoint, the shape in current model is torch.Size([128]).
        size mismatch for encoder.encoder.blocks.3.3.bn3.running_mean: copying a param with shape torch.Size([112]) from checkpoint, the shape in current model is torch.Size([128]).
        size mismatch for encoder.encoder.blocks.3.3.bn3.running_var: copying a param with shape torch.Size([112]) from checkpoint, the shape in current model is torch.Size([128]).
        size mismatch for encoder.encoder.blocks.3.4.conv_pw.weight: copying a param with shape torch.Size([672, 112, 1, 1]) from checkpoint, the shape in current model is torch.Size([768, 128, 1, 1]).
        size mismatch for encoder.encoder.blocks.3.4.bn1.weight: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for encoder.encoder.blocks.3.4.bn1.bias: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for encoder.encoder.blocks.3.4.bn1.running_mean: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for encoder.encoder.blocks.3.4.bn1.running_var: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for encoder.encoder.blocks.3.4.conv_dw.weight: copying a param with shape torch.Size([672, 1, 3, 3]) from checkpoint, the shape in current model is torch.Size([768, 1, 3, 3]).
        size mismatch for encoder.encoder.blocks.3.4.bn2.weight: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for encoder.encoder.blocks.3.4.bn2.bias: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for encoder.encoder.blocks.3.4.bn2.running_mean: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for encoder.encoder.blocks.3.4.bn2.running_var: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for encoder.encoder.blocks.3.4.se.conv_reduce.weight: copying a param with shape torch.Size([28, 672, 1, 1]) from checkpoint, the shape in current model is torch.Size([32, 768, 1, 1]).
        size mismatch for encoder.encoder.blocks.3.4.se.conv_reduce.bias: copying a param with shape torch.Size([28]) from checkpoint, the shape in current model is torch.Size([32]).
        size mismatch for encoder.encoder.blocks.3.4.se.conv_expand.weight: copying a param with shape torch.Size([672, 28, 1, 1]) from checkpoint, the shape in current model is torch.Size([768, 32, 1, 1]).
        size mismatch for encoder.encoder.blocks.3.4.se.conv_expand.bias: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for encoder.encoder.blocks.3.4.conv_pwl.weight: copying a param with shape torch.Size([112, 672, 1, 1]) from checkpoint, the shape in current model is torch.Size([128, 768, 1, 1]).
        size mismatch for encoder.encoder.blocks.3.4.bn3.weight: copying a param with shape torch.Size([112]) from checkpoint, the shape in current model is torch.Size([128]).
        size mismatch for encoder.encoder.blocks.3.4.bn3.bias: copying a param with shape torch.Size([112]) from checkpoint, the shape in current model is torch.Size([128]).
        size mismatch for encoder.encoder.blocks.3.4.bn3.running_mean: copying a param with shape torch.Size([112]) from checkpoint, the shape in current model is torch.Size([128]).
        size mismatch for encoder.encoder.blocks.3.4.bn3.running_var: copying a param with shape torch.Size([112]) from checkpoint, the shape in current model is torch.Size([128]).
        size mismatch for encoder.encoder.blocks.3.5.conv_pw.weight: copying a param with shape torch.Size([672, 112, 1, 1]) from checkpoint, the shape in current model is torch.Size([768, 128, 1, 1]).
        size mismatch for encoder.encoder.blocks.3.5.bn1.weight: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for encoder.encoder.blocks.3.5.bn1.bias: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for encoder.encoder.blocks.3.5.bn1.running_mean: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for encoder.encoder.blocks.3.5.bn1.running_var: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for encoder.encoder.blocks.3.5.conv_dw.weight: copying a param with shape torch.Size([672, 1, 3, 3]) from checkpoint, the shape in current model is torch.Size([768, 1, 3, 3]).
        size mismatch for encoder.encoder.blocks.3.5.bn2.weight: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for encoder.encoder.blocks.3.5.bn2.bias: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for encoder.encoder.blocks.3.5.bn2.running_mean: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for encoder.encoder.blocks.3.5.bn2.running_var: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for encoder.encoder.blocks.3.5.se.conv_reduce.weight: copying a param with shape torch.Size([28, 672, 1, 1]) from checkpoint, the shape in current model is torch.Size([32, 768, 1, 1]).
        size mismatch for encoder.encoder.blocks.3.5.se.conv_reduce.bias: copying a param with shape torch.Size([28]) from checkpoint, the shape in current model is torch.Size([32]).
        size mismatch for encoder.encoder.blocks.3.5.se.conv_expand.weight: copying a param with shape torch.Size([672, 28, 1, 1]) from checkpoint, the shape in current model is torch.Size([768, 32, 1, 1]).
        size mismatch for encoder.encoder.blocks.3.5.se.conv_expand.bias: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for encoder.encoder.blocks.3.5.conv_pwl.weight: copying a param with shape torch.Size([112, 672, 1, 1]) from checkpoint, the shape in current model is torch.Size([128, 768, 1, 1]).
        size mismatch for encoder.encoder.blocks.3.5.bn3.weight: copying a param with shape torch.Size([112]) from checkpoint, the shape in current model is torch.Size([128]).
        size mismatch for encoder.encoder.blocks.3.5.bn3.bias: copying a param with shape torch.Size([112]) from checkpoint, the shape in current model is torch.Size([128]).
        size mismatch for encoder.encoder.blocks.3.5.bn3.running_mean: copying a param with shape torch.Size([112]) from checkpoint, the shape in current model is torch.Size([128]).
        size mismatch for encoder.encoder.blocks.3.5.bn3.running_var: copying a param with shape torch.Size([112]) from checkpoint, the shape in current model is torch.Size([128]).
        size mismatch for encoder.encoder.blocks.4.0.conv_pw.weight: copying a param with shape torch.Size([672, 112, 1, 1]) from checkpoint, the shape in current model is torch.Size([768, 128, 1, 1]).
        size mismatch for encoder.encoder.blocks.4.0.bn1.weight: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for encoder.encoder.blocks.4.0.bn1.bias: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for encoder.encoder.blocks.4.0.bn1.running_mean: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for encoder.encoder.blocks.4.0.bn1.running_var: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for encoder.encoder.blocks.4.0.conv_dw.weight: copying a param with shape torch.Size([672, 1, 5, 5]) from checkpoint, the shape in current model is torch.Size([768, 1, 5, 5]).
        size mismatch for encoder.encoder.blocks.4.0.bn2.weight: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for encoder.encoder.blocks.4.0.bn2.bias: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for encoder.encoder.blocks.4.0.bn2.running_mean: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for encoder.encoder.blocks.4.0.bn2.running_var: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for encoder.encoder.blocks.4.0.se.conv_reduce.weight: copying a param with shape torch.Size([28, 672, 1, 1]) from checkpoint, the shape in current model is torch.Size([32, 768, 1, 1]).
        size mismatch for encoder.encoder.blocks.4.0.se.conv_reduce.bias: copying a param with shape torch.Size([28]) from checkpoint, the shape in current model is torch.Size([32]).
        size mismatch for encoder.encoder.blocks.4.0.se.conv_expand.weight: copying a param with shape torch.Size([672, 28, 1, 1]) from checkpoint, the shape in current model is torch.Size([768, 32, 1, 1]).
        size mismatch for encoder.encoder.blocks.4.0.se.conv_expand.bias: copying a param with shape torch.Size([672]) from checkpoint, the shape in current model is torch.Size([768]).
        size mismatch for encoder.encoder.blocks.4.0.conv_pwl.weight: copying a param with shape torch.Size([160, 672, 1, 1]) from checkpoint, the shape in current model is torch.Size([176, 768, 1, 1]).
        size mismatch for encoder.encoder.blocks.4.0.bn3.weight: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([176]).
        size mismatch for encoder.encoder.blocks.4.0.bn3.bias: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([176]).
        size mismatch for encoder.encoder.blocks.4.0.bn3.running_mean: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([176]).
        size mismatch for encoder.encoder.blocks.4.0.bn3.running_var: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([176]).
        size mismatch for encoder.encoder.blocks.4.1.conv_pw.weight: copying a param with shape torch.Size([960, 160, 1, 1]) from checkpoint, the shape in current model is torch.Size([1056, 176, 1, 1]).
        size mismatch for encoder.encoder.blocks.4.1.bn1.weight: copying a param with shape torch.Size([960]) from checkpoint, the shape in current model is torch.Size([1056]).
        size mismatch for encoder.encoder.blocks.4.1.bn1.bias: copying a param with shape torch.Size([960]) from checkpoint, the shape in current model is torch.Size([1056]).
        size mismatch for encoder.encoder.blocks.4.1.bn1.running_mean: copying a param with shape torch.Size([960]) from checkpoint, the shape in current model is torch.Size([1056]).
        size mismatch for encoder.encoder.blocks.4.1.bn1.running_var: copying a param with shape torch.Size([960]) from checkpoint, the shape in current model is torch.Size([1056]).
        size mismatch for encoder.encoder.blocks.4.1.conv_dw.weight: copying a param with shape torch.Size([960, 1, 5, 5]) from checkpoint, the shape in current model is torch.Size([1056, 1, 5, 5]).
        size mismatch for encoder.encoder.blocks.4.1.bn2.weight: copying a param with shape torch.Size([960]) from checkpoint, the shape in current model is torch.Size([1056]).
        size mismatch for encoder.encoder.blocks.4.1.bn2.bias: copying a param with shape torch.Size([960]) from checkpoint, the shape in current model is torch.Size([1056]).
        size mismatch for encoder.encoder.blocks.4.1.bn2.running_mean: copying a param with shape torch.Size([960]) from checkpoint, the shape in current model is torch.Size([1056]).
        size mismatch for encoder.encoder.blocks.4.1.bn2.running_var: copying a param with shape torch.Size([960]) from checkpoint, the shape in current model is torch.Size([1056]).
        size mismatch for encoder.encoder.blocks.4.1.se.conv_reduce.weight: copying a param with shape torch.Size([40, 960, 1, 1]) from checkpoint, the shape in current model is torch.Size([44, 1056, 1, 1]).
        size mismatch for encoder.encoder.blocks.4.1.se.conv_reduce.bias: copying a param with shape torch.Size([40]) from checkpoint, the shape in current model is torch.Size([44]).
        size mismatch for encoder.encoder.blocks.4.1.se.conv_expand.weight: copying a param with shape torch.Size([960, 40, 1, 1]) from checkpoint, the shape in current model is torch.Size([1056, 44, 1, 1]).
        size mismatch for encoder.encoder.blocks.4.1.se.conv_expand.bias: copying a param with shape torch.Size([960]) from checkpoint, the shape in current model is torch.Size([1056]).
        size mismatch for encoder.encoder.blocks.4.1.conv_pwl.weight: copying a param with shape torch.Size([160, 960, 1, 1]) from checkpoint, the shape in current model is torch.Size([176, 1056, 1, 1]).
        size mismatch for encoder.encoder.blocks.4.1.bn3.weight: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([176]).
        size mismatch for encoder.encoder.blocks.4.1.bn3.bias: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([176]).
        size mismatch for encoder.encoder.blocks.4.1.bn3.running_mean: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([176]).
        size mismatch for encoder.encoder.blocks.4.1.bn3.running_var: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([176]).
        size mismatch for encoder.encoder.blocks.4.2.conv_pw.weight: copying a param with shape torch.Size([960, 160, 1, 1]) from checkpoint, the shape in current model is torch.Size([1056, 176, 1, 1]).
        size mismatch for encoder.encoder.blocks.4.2.bn1.weight: copying a param with shape torch.Size([960]) from checkpoint, the shape in current model is torch.Size([1056]).
        size mismatch for encoder.encoder.blocks.4.2.bn1.bias: copying a param with shape torch.Size([960]) from checkpoint, the shape in current model is torch.Size([1056]).
        size mismatch for encoder.encoder.blocks.4.2.bn1.running_mean: copying a param with shape torch.Size([960]) from checkpoint, the shape in current model is torch.Size([1056]).
        size mismatch for encoder.encoder.blocks.4.2.bn1.running_var: copying a param with shape torch.Size([960]) from checkpoint, the shape in current model is torch.Size([1056]).
        size mismatch for encoder.encoder.blocks.4.2.conv_dw.weight: copying a param with shape torch.Size([960, 1, 5, 5]) from checkpoint, the shape in current model is torch.Size([1056, 1, 5, 5]).
        size mismatch for encoder.encoder.blocks.4.2.bn2.weight: copying a param with shape torch.Size([960]) from checkpoint, the shape in current model is torch.Size([1056]).
        size mismatch for encoder.encoder.blocks.4.2.bn2.bias: copying a param with shape torch.Size([960]) from checkpoint, the shape in current model is torch.Size([1056]).
        size mismatch for encoder.encoder.blocks.4.2.bn2.running_mean: copying a param with shape torch.Size([960]) from checkpoint, the shape in current model is torch.Size([1056]).
        size mismatch for encoder.encoder.blocks.4.2.bn2.running_var: copying a param with shape torch.Size([960]) from checkpoint, the shape in current model is torch.Size([1056]).
        size mismatch for encoder.encoder.blocks.4.2.se.conv_reduce.weight: copying a param with shape torch.Size([40, 960, 1, 1]) from checkpoint, the shape in current model is torch.Size([44, 1056, 1, 1]).
        size mismatch for encoder.encoder.blocks.4.2.se.conv_reduce.bias: copying a param with shape torch.Size([40]) from checkpoint, the shape in current model is torch.Size([44]).
        size mismatch for encoder.encoder.blocks.4.2.se.conv_expand.weight: copying a param with shape torch.Size([960, 40, 1, 1]) from checkpoint, the shape in current model is torch.Size([1056, 44, 1, 1]).
        size mismatch for encoder.encoder.blocks.4.2.se.conv_expand.bias: copying a param with shape torch.Size([960]) from checkpoint, the shape in current model is torch.Size([1056]).
        size mismatch for encoder.encoder.blocks.4.2.conv_pwl.weight: copying a param with shape torch.Size([160, 960, 1, 1]) from checkpoint, the shape in current model is torch.Size([176, 1056, 1, 1]).
        size mismatch for encoder.encoder.blocks.4.2.bn3.weight: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([176]).
        size mismatch for encoder.encoder.blocks.4.2.bn3.bias: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([176]).
        size mismatch for encoder.encoder.blocks.4.2.bn3.running_mean: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([176]).
        size mismatch for encoder.encoder.blocks.4.2.bn3.running_var: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([176]).
        size mismatch for encoder.encoder.blocks.4.3.conv_pw.weight: copying a param with shape torch.Size([960, 160, 1, 1]) from checkpoint, the shape in current model is torch.Size([1056, 176, 1, 1]).
        size mismatch for encoder.encoder.blocks.4.3.bn1.weight: copying a param with shape torch.Size([960]) from checkpoint, the shape in current model is torch.Size([1056]).
        size mismatch for encoder.encoder.blocks.4.3.bn1.bias: copying a param with shape torch.Size([960]) from checkpoint, the shape in current model is torch.Size([1056]).
        size mismatch for encoder.encoder.blocks.4.3.bn1.running_mean: copying a param with shape torch.Size([960]) from checkpoint, the shape in current model is torch.Size([1056]).
        size mismatch for encoder.encoder.blocks.4.3.bn1.running_var: copying a param with shape torch.Size([960]) from checkpoint, the shape in current model is torch.Size([1056]).
        size mismatch for encoder.encoder.blocks.4.3.conv_dw.weight: copying a param with shape torch.Size([960, 1, 5, 5]) from checkpoint, the shape in current model is torch.Size([1056, 1, 5, 5]).
        size mismatch for encoder.encoder.blocks.4.3.bn2.weight: copying a param with shape torch.Size([960]) from checkpoint, the shape in current model is torch.Size([1056]).
        size mismatch for encoder.encoder.blocks.4.3.bn2.bias: copying a param with shape torch.Size([960]) from checkpoint, the shape in current model is torch.Size([1056]).
        size mismatch for encoder.encoder.blocks.4.3.bn2.running_mean: copying a param with shape torch.Size([960]) from checkpoint, the shape in current model is torch.Size([1056]).
        size mismatch for encoder.encoder.blocks.4.3.bn2.running_var: copying a param with shape torch.Size([960]) from checkpoint, the shape in current model is torch.Size([1056]).
        size mismatch for encoder.encoder.blocks.4.3.se.conv_reduce.weight: copying a param with shape torch.Size([40, 960, 1, 1]) from checkpoint, the shape in current model is torch.Size([44, 1056, 1, 1]).
        size mismatch for encoder.encoder.blocks.4.3.se.conv_reduce.bias: copying a param with shape torch.Size([40]) from checkpoint, the shape in current model is torch.Size([44]).
        size mismatch for encoder.encoder.blocks.4.3.se.conv_expand.weight: copying a param with shape torch.Size([960, 40, 1, 1]) from checkpoint, the shape in current model is torch.Size([1056, 44, 1, 1]).
        size mismatch for encoder.encoder.blocks.4.3.se.conv_expand.bias: copying a param with shape torch.Size([960]) from checkpoint, the shape in current model is torch.Size([1056]).
        size mismatch for encoder.encoder.blocks.4.3.conv_pwl.weight: copying a param with shape torch.Size([160, 960, 1, 1]) from checkpoint, the shape in current model is torch.Size([176, 1056, 1, 1]).
        size mismatch for encoder.encoder.blocks.4.3.bn3.weight: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([176]).
        size mismatch for encoder.encoder.blocks.4.3.bn3.bias: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([176]).
        size mismatch for encoder.encoder.blocks.4.3.bn3.running_mean: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([176]).
        size mismatch for encoder.encoder.blocks.4.3.bn3.running_var: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([176]).
        size mismatch for encoder.encoder.blocks.4.4.conv_pw.weight: copying a param with shape torch.Size([960, 160, 1, 1]) from checkpoint, the shape in current model is torch.Size([1056, 176, 1, 1]).
        size mismatch for encoder.encoder.blocks.4.4.bn1.weight: copying a param with shape torch.Size([960]) from checkpoint, the shape in current model is torch.Size([1056]).
        size mismatch for encoder.encoder.blocks.4.4.bn1.bias: copying a param with shape torch.Size([960]) from checkpoint, the shape in current model is torch.Size([1056]).
        size mismatch for encoder.encoder.blocks.4.4.bn1.running_mean: copying a param with shape torch.Size([960]) from checkpoint, the shape in current model is torch.Size([1056]).
        size mismatch for encoder.encoder.blocks.4.4.bn1.running_var: copying a param with shape torch.Size([960]) from checkpoint, the shape in current model is torch.Size([1056]).
        size mismatch for encoder.encoder.blocks.4.4.conv_dw.weight: copying a param with shape torch.Size([960, 1, 5, 5]) from checkpoint, the shape in current model is torch.Size([1056, 1, 5, 5]).
        size mismatch for encoder.encoder.blocks.4.4.bn2.weight: copying a param with shape torch.Size([960]) from checkpoint, the shape in current model is torch.Size([1056]).
        size mismatch for encoder.encoder.blocks.4.4.bn2.bias: copying a param with shape torch.Size([960]) from checkpoint, the shape in current model is torch.Size([1056]).
        size mismatch for encoder.encoder.blocks.4.4.bn2.running_mean: copying a param with shape torch.Size([960]) from checkpoint, the shape in current model is torch.Size([1056]).
        size mismatch for encoder.encoder.blocks.4.4.bn2.running_var: copying a param with shape torch.Size([960]) from checkpoint, the shape in current model is torch.Size([1056]).
        size mismatch for encoder.encoder.blocks.4.4.se.conv_reduce.weight: copying a param with shape torch.Size([40, 960, 1, 1]) from checkpoint, the shape in current model is torch.Size([44, 1056, 1, 1]).
        size mismatch for encoder.encoder.blocks.4.4.se.conv_reduce.bias: copying a param with shape torch.Size([40]) from checkpoint, the shape in current model is torch.Size([44]).
        size mismatch for encoder.encoder.blocks.4.4.se.conv_expand.weight: copying a param with shape torch.Size([960, 40, 1, 1]) from checkpoint, the shape in current model is torch.Size([1056, 44, 1, 1]).
        size mismatch for encoder.encoder.blocks.4.4.se.conv_expand.bias: copying a param with shape torch.Size([960]) from checkpoint, the shape in current model is torch.Size([1056]).
        size mismatch for encoder.encoder.blocks.4.4.conv_pwl.weight: copying a param with shape torch.Size([160, 960, 1, 1]) from checkpoint, the shape in current model is torch.Size([176, 1056, 1, 1]).
        size mismatch for encoder.encoder.blocks.4.4.bn3.weight: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([176]).
        size mismatch for encoder.encoder.blocks.4.4.bn3.bias: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([176]).
        size mismatch for encoder.encoder.blocks.4.4.bn3.running_mean: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([176]).
        size mismatch for encoder.encoder.blocks.4.4.bn3.running_var: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([176]).
        size mismatch for encoder.encoder.blocks.4.5.conv_pw.weight: copying a param with shape torch.Size([960, 160, 1, 1]) from checkpoint, the shape in current model is torch.Size([1056, 176, 1, 1]).
        size mismatch for encoder.encoder.blocks.4.5.bn1.weight: copying a param with shape torch.Size([960]) from checkpoint, the shape in current model is torch.Size([1056]).
        size mismatch for encoder.encoder.blocks.4.5.bn1.bias: copying a param with shape torch.Size([960]) from checkpoint, the shape in current model is torch.Size([1056]).
        size mismatch for encoder.encoder.blocks.4.5.bn1.running_mean: copying a param with shape torch.Size([960]) from checkpoint, the shape in current model is torch.Size([1056]).
        size mismatch for encoder.encoder.blocks.4.5.bn1.running_var: copying a param with shape torch.Size([960]) from checkpoint, the shape in current model is torch.Size([1056]).
        size mismatch for encoder.encoder.blocks.4.5.conv_dw.weight: copying a param with shape torch.Size([960, 1, 5, 5]) from checkpoint, the shape in current model is torch.Size([1056, 1, 5, 5]).
        size mismatch for encoder.encoder.blocks.4.5.bn2.weight: copying a param with shape torch.Size([960]) from checkpoint, the shape in current model is torch.Size([1056]).
        size mismatch for encoder.encoder.blocks.4.5.bn2.bias: copying a param with shape torch.Size([960]) from checkpoint, the shape in current model is torch.Size([1056]).
        size mismatch for encoder.encoder.blocks.4.5.bn2.running_mean: copying a param with shape torch.Size([960]) from checkpoint, the shape in current model is torch.Size([1056]).
        size mismatch for encoder.encoder.blocks.4.5.bn2.running_var: copying a param with shape torch.Size([960]) from checkpoint, the shape in current model is torch.Size([1056]).
        size mismatch for encoder.encoder.blocks.4.5.se.conv_reduce.weight: copying a param with shape torch.Size([40, 960, 1, 1]) from checkpoint, the shape in current model is torch.Size([44, 1056, 1, 1]).
        size mismatch for encoder.encoder.blocks.4.5.se.conv_reduce.bias: copying a param with shape torch.Size([40]) from checkpoint, the shape in current model is torch.Size([44]).
        size mismatch for encoder.encoder.blocks.4.5.se.conv_expand.weight: copying a param with shape torch.Size([960, 40, 1, 1]) from checkpoint, the shape in current model is torch.Size([1056, 44, 1, 1]).
        size mismatch for encoder.encoder.blocks.4.5.se.conv_expand.bias: copying a param with shape torch.Size([960]) from checkpoint, the shape in current model is torch.Size([1056]).
        size mismatch for encoder.encoder.blocks.4.5.conv_pwl.weight: copying a param with shape torch.Size([160, 960, 1, 1]) from checkpoint, the shape in current model is torch.Size([176, 1056, 1, 1]).
        size mismatch for encoder.encoder.blocks.4.5.bn3.weight: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([176]).
        size mismatch for encoder.encoder.blocks.4.5.bn3.bias: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([176]).
        size mismatch for encoder.encoder.blocks.4.5.bn3.running_mean: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([176]).
        size mismatch for encoder.encoder.blocks.4.5.bn3.running_var: copying a param with shape torch.Size([160]) from checkpoint, the shape in current model is torch.Size([176]).
        size mismatch for encoder.encoder.blocks.5.0.conv_pw.weight: copying a param with shape torch.Size([960, 160, 1, 1]) from checkpoint, the shape in current model is torch.Size([1056, 176, 1, 1]).
        size mismatch for encoder.encoder.blocks.5.0.bn1.weight: copying a param with shape torch.Size([960]) from checkpoint, the shape in current model is torch.Size([1056]).
        size mismatch for encoder.encoder.blocks.5.0.bn1.bias: copying a param with shape torch.Size([960]) from checkpoint, the shape in current model is torch.Size([1056]).
        size mismatch for encoder.encoder.blocks.5.0.bn1.running_mean: copying a param with shape torch.Size([960]) from checkpoint, the shape in current model is torch.Size([1056]).
        size mismatch for encoder.encoder.blocks.5.0.bn1.running_var: copying a param with shape torch.Size([960]) from checkpoint, the shape in current model is torch.Size([1056]).
        size mismatch for encoder.encoder.blocks.5.0.conv_dw.weight: copying a param with shape torch.Size([960, 1, 5, 5]) from checkpoint, the shape in current model is torch.Size([1056, 1, 5, 5]).
        size mismatch for encoder.encoder.blocks.5.0.bn2.weight: copying a param with shape torch.Size([960]) from checkpoint, the shape in current model is torch.Size([1056]).
        size mismatch for encoder.encoder.blocks.5.0.bn2.bias: copying a param with shape torch.Size([960]) from checkpoint, the shape in current model is torch.Size([1056]).
        size mismatch for encoder.encoder.blocks.5.0.bn2.running_mean: copying a param with shape torch.Size([960]) from checkpoint, the shape in current model is torch.Size([1056]).
        size mismatch for encoder.encoder.blocks.5.0.bn2.running_var: copying a param with shape torch.Size([960]) from checkpoint, the shape in current model is torch.Size([1056]).
        size mismatch for encoder.encoder.blocks.5.0.se.conv_reduce.weight: copying a param with shape torch.Size([40, 960, 1, 1]) from checkpoint, the shape in current model is torch.Size([44, 1056, 1, 1]).
        size mismatch for encoder.encoder.blocks.5.0.se.conv_reduce.bias: copying a param with shape torch.Size([40]) from checkpoint, the shape in current model is torch.Size([44]).
        size mismatch for encoder.encoder.blocks.5.0.se.conv_expand.weight: copying a param with shape torch.Size([960, 40, 1, 1]) from checkpoint, the shape in current model is torch.Size([1056, 44, 1, 1]).
        size mismatch for encoder.encoder.blocks.5.0.se.conv_expand.bias: copying a param with shape torch.Size([960]) from checkpoint, the shape in current model is torch.Size([1056]).
        size mismatch for encoder.encoder.blocks.5.0.conv_pwl.weight: copying a param with shape torch.Size([272, 960, 1, 1]) from checkpoint, the shape in current model is torch.Size([304, 1056, 1, 1]).
        size mismatch for encoder.encoder.blocks.5.0.bn3.weight: copying a param with shape torch.Size([272]) from checkpoint, the shape in current model is torch.Size([304]).
        size mismatch for encoder.encoder.blocks.5.0.bn3.bias: copying a param with shape torch.Size([272]) from checkpoint, the shape in current model is torch.Size([304]).
        size mismatch for encoder.encoder.blocks.5.0.bn3.running_mean: copying a param with shape torch.Size([272]) from checkpoint, the shape in current model is torch.Size([304]).
        size mismatch for encoder.encoder.blocks.5.0.bn3.running_var: copying a param with shape torch.Size([272]) from checkpoint, the shape in current model is torch.Size([304]).
        size mismatch for encoder.encoder.blocks.5.1.conv_pw.weight: copying a param with shape torch.Size([1632, 272, 1, 1]) from checkpoint, the shape in current model is torch.Size([1824, 304, 1, 1]).
        size mismatch for encoder.encoder.blocks.5.1.bn1.weight: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.5.1.bn1.bias: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.5.1.bn1.running_mean: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.5.1.bn1.running_var: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.5.1.conv_dw.weight: copying a param with shape torch.Size([1632, 1, 5, 5]) from checkpoint, the shape in current model is torch.Size([1824, 1, 5, 5]).
        size mismatch for encoder.encoder.blocks.5.1.bn2.weight: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.5.1.bn2.bias: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.5.1.bn2.running_mean: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.5.1.bn2.running_var: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.5.1.se.conv_reduce.weight: copying a param with shape torch.Size([68, 1632, 1, 1]) from checkpoint, the shape in current model is torch.Size([76, 1824, 1, 1]).
        size mismatch for encoder.encoder.blocks.5.1.se.conv_reduce.bias: copying a param with shape torch.Size([68]) from checkpoint, the shape in current model is torch.Size([76]).
        size mismatch for encoder.encoder.blocks.5.1.se.conv_expand.weight: copying a param with shape torch.Size([1632, 68, 1, 1]) from checkpoint, the shape in current model is torch.Size([1824, 76, 1, 1]).
        size mismatch for encoder.encoder.blocks.5.1.se.conv_expand.bias: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.5.1.conv_pwl.weight: copying a param with shape torch.Size([272, 1632, 1, 1]) from checkpoint, the shape in current model is torch.Size([304, 1824, 1, 1]).
        size mismatch for encoder.encoder.blocks.5.1.bn3.weight: copying a param with shape torch.Size([272]) from checkpoint, the shape in current model is torch.Size([304]).
        size mismatch for encoder.encoder.blocks.5.1.bn3.bias: copying a param with shape torch.Size([272]) from checkpoint, the shape in current model is torch.Size([304]).
        size mismatch for encoder.encoder.blocks.5.1.bn3.running_mean: copying a param with shape torch.Size([272]) from checkpoint, the shape in current model is torch.Size([304]).
        size mismatch for encoder.encoder.blocks.5.1.bn3.running_var: copying a param with shape torch.Size([272]) from checkpoint, the shape in current model is torch.Size([304]).
        size mismatch for encoder.encoder.blocks.5.2.conv_pw.weight: copying a param with shape torch.Size([1632, 272, 1, 1]) from checkpoint, the shape in current model is torch.Size([1824, 304, 1, 1]).
        size mismatch for encoder.encoder.blocks.5.2.bn1.weight: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.5.2.bn1.bias: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.5.2.bn1.running_mean: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.5.2.bn1.running_var: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.5.2.conv_dw.weight: copying a param with shape torch.Size([1632, 1, 5, 5]) from checkpoint, the shape in current model is torch.Size([1824, 1, 5, 5]).
        size mismatch for encoder.encoder.blocks.5.2.bn2.weight: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.5.2.bn2.bias: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.5.2.bn2.running_mean: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.5.2.bn2.running_var: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.5.2.se.conv_reduce.weight: copying a param with shape torch.Size([68, 1632, 1, 1]) from checkpoint, the shape in current model is torch.Size([76, 1824, 1, 1]).
        size mismatch for encoder.encoder.blocks.5.2.se.conv_reduce.bias: copying a param with shape torch.Size([68]) from checkpoint, the shape in current model is torch.Size([76]).
        size mismatch for encoder.encoder.blocks.5.2.se.conv_expand.weight: copying a param with shape torch.Size([1632, 68, 1, 1]) from checkpoint, the shape in current model is torch.Size([1824, 76, 1, 1]).
        size mismatch for encoder.encoder.blocks.5.2.se.conv_expand.bias: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.5.2.conv_pwl.weight: copying a param with shape torch.Size([272, 1632, 1, 1]) from checkpoint, the shape in current model is torch.Size([304, 1824, 1, 1]).
        size mismatch for encoder.encoder.blocks.5.2.bn3.weight: copying a param with shape torch.Size([272]) from checkpoint, the shape in current model is torch.Size([304]).
        size mismatch for encoder.encoder.blocks.5.2.bn3.bias: copying a param with shape torch.Size([272]) from checkpoint, the shape in current model is torch.Size([304]).
        size mismatch for encoder.encoder.blocks.5.2.bn3.running_mean: copying a param with shape torch.Size([272]) from checkpoint, the shape in current model is torch.Size([304]).
        size mismatch for encoder.encoder.blocks.5.2.bn3.running_var: copying a param with shape torch.Size([272]) from checkpoint, the shape in current model is torch.Size([304]).
        size mismatch for encoder.encoder.blocks.5.3.conv_pw.weight: copying a param with shape torch.Size([1632, 272, 1, 1]) from checkpoint, the shape in current model is torch.Size([1824, 304, 1, 1]).
        size mismatch for encoder.encoder.blocks.5.3.bn1.weight: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.5.3.bn1.bias: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.5.3.bn1.running_mean: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.5.3.bn1.running_var: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.5.3.conv_dw.weight: copying a param with shape torch.Size([1632, 1, 5, 5]) from checkpoint, the shape in current model is torch.Size([1824, 1, 5, 5]).
        size mismatch for encoder.encoder.blocks.5.3.bn2.weight: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.5.3.bn2.bias: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.5.3.bn2.running_mean: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.5.3.bn2.running_var: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.5.3.se.conv_reduce.weight: copying a param with shape torch.Size([68, 1632, 1, 1]) from checkpoint, the shape in current model is torch.Size([76, 1824, 1, 1]).
        size mismatch for encoder.encoder.blocks.5.3.se.conv_reduce.bias: copying a param with shape torch.Size([68]) from checkpoint, the shape in current model is torch.Size([76]).
        size mismatch for encoder.encoder.blocks.5.3.se.conv_expand.weight: copying a param with shape torch.Size([1632, 68, 1, 1]) from checkpoint, the shape in current model is torch.Size([1824, 76, 1, 1]).
        size mismatch for encoder.encoder.blocks.5.3.se.conv_expand.bias: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.5.3.conv_pwl.weight: copying a param with shape torch.Size([272, 1632, 1, 1]) from checkpoint, the shape in current model is torch.Size([304, 1824, 1, 1]).
        size mismatch for encoder.encoder.blocks.5.3.bn3.weight: copying a param with shape torch.Size([272]) from checkpoint, the shape in current model is torch.Size([304]).
        size mismatch for encoder.encoder.blocks.5.3.bn3.bias: copying a param with shape torch.Size([272]) from checkpoint, the shape in current model is torch.Size([304]).
        size mismatch for encoder.encoder.blocks.5.3.bn3.running_mean: copying a param with shape torch.Size([272]) from checkpoint, the shape in current model is torch.Size([304]).
        size mismatch for encoder.encoder.blocks.5.3.bn3.running_var: copying a param with shape torch.Size([272]) from checkpoint, the shape in current model is torch.Size([304]).
        size mismatch for encoder.encoder.blocks.5.4.conv_pw.weight: copying a param with shape torch.Size([1632, 272, 1, 1]) from checkpoint, the shape in current model is torch.Size([1824, 304, 1, 1]).
        size mismatch for encoder.encoder.blocks.5.4.bn1.weight: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.5.4.bn1.bias: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.5.4.bn1.running_mean: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.5.4.bn1.running_var: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.5.4.conv_dw.weight: copying a param with shape torch.Size([1632, 1, 5, 5]) from checkpoint, the shape in current model is torch.Size([1824, 1, 5, 5]).
        size mismatch for encoder.encoder.blocks.5.4.bn2.weight: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.5.4.bn2.bias: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.5.4.bn2.running_mean: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.5.4.bn2.running_var: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.5.4.se.conv_reduce.weight: copying a param with shape torch.Size([68, 1632, 1, 1]) from checkpoint, the shape in current model is torch.Size([76, 1824, 1, 1]).
        size mismatch for encoder.encoder.blocks.5.4.se.conv_reduce.bias: copying a param with shape torch.Size([68]) from checkpoint, the shape in current model is torch.Size([76]).
        size mismatch for encoder.encoder.blocks.5.4.se.conv_expand.weight: copying a param with shape torch.Size([1632, 68, 1, 1]) from checkpoint, the shape in current model is torch.Size([1824, 76, 1, 1]).
        size mismatch for encoder.encoder.blocks.5.4.se.conv_expand.bias: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.5.4.conv_pwl.weight: copying a param with shape torch.Size([272, 1632, 1, 1]) from checkpoint, the shape in current model is torch.Size([304, 1824, 1, 1]).
        size mismatch for encoder.encoder.blocks.5.4.bn3.weight: copying a param with shape torch.Size([272]) from checkpoint, the shape in current model is torch.Size([304]).
        size mismatch for encoder.encoder.blocks.5.4.bn3.bias: copying a param with shape torch.Size([272]) from checkpoint, the shape in current model is torch.Size([304]).
        size mismatch for encoder.encoder.blocks.5.4.bn3.running_mean: copying a param with shape torch.Size([272]) from checkpoint, the shape in current model is torch.Size([304]).
        size mismatch for encoder.encoder.blocks.5.4.bn3.running_var: copying a param with shape torch.Size([272]) from checkpoint, the shape in current model is torch.Size([304]).
        size mismatch for encoder.encoder.blocks.5.5.conv_pw.weight: copying a param with shape torch.Size([1632, 272, 1, 1]) from checkpoint, the shape in current model is torch.Size([1824, 304, 1, 1]).
        size mismatch for encoder.encoder.blocks.5.5.bn1.weight: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.5.5.bn1.bias: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.5.5.bn1.running_mean: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.5.5.bn1.running_var: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.5.5.conv_dw.weight: copying a param with shape torch.Size([1632, 1, 5, 5]) from checkpoint, the shape in current model is torch.Size([1824, 1, 5, 5]).
        size mismatch for encoder.encoder.blocks.5.5.bn2.weight: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.5.5.bn2.bias: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.5.5.bn2.running_mean: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.5.5.bn2.running_var: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.5.5.se.conv_reduce.weight: copying a param with shape torch.Size([68, 1632, 1, 1]) from checkpoint, the shape in current model is torch.Size([76, 1824, 1, 1]).
        size mismatch for encoder.encoder.blocks.5.5.se.conv_reduce.bias: copying a param with shape torch.Size([68]) from checkpoint, the shape in current model is torch.Size([76]).
        size mismatch for encoder.encoder.blocks.5.5.se.conv_expand.weight: copying a param with shape torch.Size([1632, 68, 1, 1]) from checkpoint, the shape in current model is torch.Size([1824, 76, 1, 1]).
        size mismatch for encoder.encoder.blocks.5.5.se.conv_expand.bias: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.5.5.conv_pwl.weight: copying a param with shape torch.Size([272, 1632, 1, 1]) from checkpoint, the shape in current model is torch.Size([304, 1824, 1, 1]).
        size mismatch for encoder.encoder.blocks.5.5.bn3.weight: copying a param with shape torch.Size([272]) from checkpoint, the shape in current model is torch.Size([304]).
        size mismatch for encoder.encoder.blocks.5.5.bn3.bias: copying a param with shape torch.Size([272]) from checkpoint, the shape in current model is torch.Size([304]).
        size mismatch for encoder.encoder.blocks.5.5.bn3.running_mean: copying a param with shape torch.Size([272]) from checkpoint, the shape in current model is torch.Size([304]).
        size mismatch for encoder.encoder.blocks.5.5.bn3.running_var: copying a param with shape torch.Size([272]) from checkpoint, the shape in current model is torch.Size([304]).
        size mismatch for encoder.encoder.blocks.5.6.conv_pw.weight: copying a param with shape torch.Size([1632, 272, 1, 1]) from checkpoint, the shape in current model is torch.Size([1824, 304, 1, 1]).
        size mismatch for encoder.encoder.blocks.5.6.bn1.weight: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.5.6.bn1.bias: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.5.6.bn1.running_mean: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.5.6.bn1.running_var: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.5.6.conv_dw.weight: copying a param with shape torch.Size([1632, 1, 5, 5]) from checkpoint, the shape in current model is torch.Size([1824, 1, 5, 5]).
        size mismatch for encoder.encoder.blocks.5.6.bn2.weight: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.5.6.bn2.bias: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.5.6.bn2.running_mean: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.5.6.bn2.running_var: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.5.6.se.conv_reduce.weight: copying a param with shape torch.Size([68, 1632, 1, 1]) from checkpoint, the shape in current model is torch.Size([76, 1824, 1, 1]).
        size mismatch for encoder.encoder.blocks.5.6.se.conv_reduce.bias: copying a param with shape torch.Size([68]) from checkpoint, the shape in current model is torch.Size([76]).
        size mismatch for encoder.encoder.blocks.5.6.se.conv_expand.weight: copying a param with shape torch.Size([1632, 68, 1, 1]) from checkpoint, the shape in current model is torch.Size([1824, 76, 1, 1]).
        size mismatch for encoder.encoder.blocks.5.6.se.conv_expand.bias: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.5.6.conv_pwl.weight: copying a param with shape torch.Size([272, 1632, 1, 1]) from checkpoint, the shape in current model is torch.Size([304, 1824, 1, 1]).
        size mismatch for encoder.encoder.blocks.5.6.bn3.weight: copying a param with shape torch.Size([272]) from checkpoint, the shape in current model is torch.Size([304]).
        size mismatch for encoder.encoder.blocks.5.6.bn3.bias: copying a param with shape torch.Size([272]) from checkpoint, the shape in current model is torch.Size([304]).
        size mismatch for encoder.encoder.blocks.5.6.bn3.running_mean: copying a param with shape torch.Size([272]) from checkpoint, the shape in current model is torch.Size([304]).
        size mismatch for encoder.encoder.blocks.5.6.bn3.running_var: copying a param with shape torch.Size([272]) from checkpoint, the shape in current model is torch.Size([304]).
        size mismatch for encoder.encoder.blocks.5.7.conv_pw.weight: copying a param with shape torch.Size([1632, 272, 1, 1]) from checkpoint, the shape in current model is torch.Size([1824, 304, 1, 1]).
        size mismatch for encoder.encoder.blocks.5.7.bn1.weight: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.5.7.bn1.bias: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.5.7.bn1.running_mean: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.5.7.bn1.running_var: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.5.7.conv_dw.weight: copying a param with shape torch.Size([1632, 1, 5, 5]) from checkpoint, the shape in current model is torch.Size([1824, 1, 5, 5]).
        size mismatch for encoder.encoder.blocks.5.7.bn2.weight: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.5.7.bn2.bias: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.5.7.bn2.running_mean: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.5.7.bn2.running_var: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.5.7.se.conv_reduce.weight: copying a param with shape torch.Size([68, 1632, 1, 1]) from checkpoint, the shape in current model is torch.Size([76, 1824, 1, 1]).
        size mismatch for encoder.encoder.blocks.5.7.se.conv_reduce.bias: copying a param with shape torch.Size([68]) from checkpoint, the shape in current model is torch.Size([76]).
        size mismatch for encoder.encoder.blocks.5.7.se.conv_expand.weight: copying a param with shape torch.Size([1632, 68, 1, 1]) from checkpoint, the shape in current model is torch.Size([1824, 76, 1, 1]).
        size mismatch for encoder.encoder.blocks.5.7.se.conv_expand.bias: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.5.7.conv_pwl.weight: copying a param with shape torch.Size([272, 1632, 1, 1]) from checkpoint, the shape in current model is torch.Size([304, 1824, 1, 1]).
        size mismatch for encoder.encoder.blocks.5.7.bn3.weight: copying a param with shape torch.Size([272]) from checkpoint, the shape in current model is torch.Size([304]).
        size mismatch for encoder.encoder.blocks.5.7.bn3.bias: copying a param with shape torch.Size([272]) from checkpoint, the shape in current model is torch.Size([304]).
        size mismatch for encoder.encoder.blocks.5.7.bn3.running_mean: copying a param with shape torch.Size([272]) from checkpoint, the shape in current model is torch.Size([304]).
        size mismatch for encoder.encoder.blocks.5.7.bn3.running_var: copying a param with shape torch.Size([272]) from checkpoint, the shape in current model is torch.Size([304]).
        size mismatch for encoder.encoder.blocks.6.0.conv_pw.weight: copying a param with shape torch.Size([1632, 272, 1, 1]) from checkpoint, the shape in current model is torch.Size([1824, 304, 1, 1]).
        size mismatch for encoder.encoder.blocks.6.0.bn1.weight: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.6.0.bn1.bias: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.6.0.bn1.running_mean: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.6.0.bn1.running_var: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.6.0.conv_dw.weight: copying a param with shape torch.Size([1632, 1, 3, 3]) from checkpoint, the shape in current model is torch.Size([1824, 1, 3, 3]).
        size mismatch for encoder.encoder.blocks.6.0.bn2.weight: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.6.0.bn2.bias: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.6.0.bn2.running_mean: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.6.0.bn2.running_var: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.6.0.se.conv_reduce.weight: copying a param with shape torch.Size([68, 1632, 1, 1]) from checkpoint, the shape in current model is torch.Size([76, 1824, 1, 1]).
        size mismatch for encoder.encoder.blocks.6.0.se.conv_reduce.bias: copying a param with shape torch.Size([68]) from checkpoint, the shape in current model is torch.Size([76]).
        size mismatch for encoder.encoder.blocks.6.0.se.conv_expand.weight: copying a param with shape torch.Size([1632, 68, 1, 1]) from checkpoint, the shape in current model is torch.Size([1824, 76, 1, 1]).
        size mismatch for encoder.encoder.blocks.6.0.se.conv_expand.bias: copying a param with shape torch.Size([1632]) from checkpoint, the shape in current model is torch.Size([1824]).
        size mismatch for encoder.encoder.blocks.6.0.conv_pwl.weight: copying a param with shape torch.Size([448, 1632, 1, 1]) from checkpoint, the shape in current model is torch.Size([512, 1824, 1, 1]).
        size mismatch for encoder.encoder.blocks.6.0.bn3.weight: copying a param with shape torch.Size([448]) from checkpoint, the shape in current model is torch.Size([512]).
        size mismatch for encoder.encoder.blocks.6.0.bn3.bias: copying a param with shape torch.Size([448]) from checkpoint, the shape in current model is torch.Size([512]).
        size mismatch for encoder.encoder.blocks.6.0.bn3.running_mean: copying a param with shape torch.Size([448]) from checkpoint, the shape in current model is torch.Size([512]).
        size mismatch for encoder.encoder.blocks.6.0.bn3.running_var: copying a param with shape torch.Size([448]) from checkpoint, the shape in current model is torch.Size([512]).
        size mismatch for encoder.encoder.blocks.6.1.conv_pw.weight: copying a param with shape torch.Size([2688, 448, 1, 1]) from checkpoint, the shape in current model is torch.Size([3072, 512, 1, 1]).
        size mismatch for encoder.encoder.blocks.6.1.bn1.weight: copying a param with shape torch.Size([2688]) from checkpoint, the shape in current model is torch.Size([3072]).
        size mismatch for encoder.encoder.blocks.6.1.bn1.bias: copying a param with shape torch.Size([2688]) from checkpoint, the shape in current model is torch.Size([3072]).
        size mismatch for encoder.encoder.blocks.6.1.bn1.running_mean: copying a param with shape torch.Size([2688]) from checkpoint, the shape in current model is torch.Size([3072]).
        size mismatch for encoder.encoder.blocks.6.1.bn1.running_var: copying a param with shape torch.Size([2688]) from checkpoint, the shape in current model is torch.Size([3072]).
        size mismatch for encoder.encoder.blocks.6.1.conv_dw.weight: copying a param with shape torch.Size([2688, 1, 3, 3]) from checkpoint, the shape in current model is torch.Size([3072, 1, 3, 3]).
        size mismatch for encoder.encoder.blocks.6.1.bn2.weight: copying a param with shape torch.Size([2688]) from checkpoint, the shape in current model is torch.Size([3072]).
        size mismatch for encoder.encoder.blocks.6.1.bn2.bias: copying a param with shape torch.Size([2688]) from checkpoint, the shape in current model is torch.Size([3072]).
        size mismatch for encoder.encoder.blocks.6.1.bn2.running_mean: copying a param with shape torch.Size([2688]) from checkpoint, the shape in current model is torch.Size([3072]).
        size mismatch for encoder.encoder.blocks.6.1.bn2.running_var: copying a param with shape torch.Size([2688]) from checkpoint, the shape in current model is torch.Size([3072]).
        size mismatch for encoder.encoder.blocks.6.1.se.conv_reduce.weight: copying a param with shape torch.Size([112, 2688, 1, 1]) from checkpoint, the shape in current model is torch.Size([128, 3072, 1, 1]).
        size mismatch for encoder.encoder.blocks.6.1.se.conv_reduce.bias: copying a param with shape torch.Size([112]) from checkpoint, the shape in current model is torch.Size([128]).
        size mismatch for encoder.encoder.blocks.6.1.se.conv_expand.weight: copying a param with shape torch.Size([2688, 112, 1, 1]) from checkpoint, the shape in current model is torch.Size([3072, 128, 1, 1]).
        size mismatch for encoder.encoder.blocks.6.1.se.conv_expand.bias: copying a param with shape torch.Size([2688]) from checkpoint, the shape in current model is torch.Size([3072]).
        size mismatch for encoder.encoder.blocks.6.1.conv_pwl.weight: copying a param with shape torch.Size([448, 2688, 1, 1]) from checkpoint, the shape in current model is torch.Size([512, 3072, 1, 1]).
        size mismatch for encoder.encoder.blocks.6.1.bn3.weight: copying a param with shape torch.Size([448]) from checkpoint, the shape in current model is torch.Size([512]).
        size mismatch for encoder.encoder.blocks.6.1.bn3.bias: copying a param with shape torch.Size([448]) from checkpoint, the shape in current model is torch.Size([512]).
        size mismatch for encoder.encoder.blocks.6.1.bn3.running_mean: copying a param with shape torch.Size([448]) from checkpoint, the shape in current model is torch.Size([512]).
        size mismatch for encoder.encoder.blocks.6.1.bn3.running_var: copying a param with shape torch.Size([448]) from checkpoint, the shape in current model is torch.Size([512]).
(sam3) samerkharboush@Samers-MBP SAR-MPS-OptunaNasFbnYoloTransformers %