dual pixel error

[xalteropsx]

    size mismatch for patch_embed.proj.weight: copying a param with shape torch.Size([48, 6, 3, 3]) from checkpoint, the shape in current model is torch.Size([48, 3, 3, 3]).     

{'inp_channels': 3, 'out_channels': 3, 'dim': 48, 'num_blocks': [4, 6, 6, 8], 'num_refinement_blocks': 4, 'heads': [1, 2, 4, 8], 'ffn_expansion_factor': 2.66, 'bias': False, 'LayerNorm_type': 'WithBias', 'dual_pixel_task': True}

dual_pixel_defocus_deblurring

model.load_state_dict(checkpoint['params'])                                                                                                                                 

File "Z:\software\python11\Lib\site-packages\torch\nn\modules\module.py", line 2191, in load_state_dict
raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format(
RuntimeError: Error(s) in loading state_dict for Restormer:
size mismatch for patch_embed.proj.weight: copying a param with shape torch.Size([48, 6, 3, 3]) from checkpoint, the shape in current model is torch.Size([48, 3, 3, 3]).