Error during my testing phase. I have written the testing program according to my requirement. But i got error in the that.

[Sayyed-Faizan010189]

024-12-11 09:53:44.634687
C:\Users\aist2345\AppData\Local\Packages\PythonSoftwareFoundation.Python.3.11_qbz5n2kfra8p0\LocalCache\local-packages\Python311\site-packages\torch\nn\init.py:452: UserWarning: Initializing zero-element tensors is a no-op
warnings.warn("Initializing zero-element tensors is a no-op")
Traceback (most recent call last):
File "c:\Users\aist2345\Documents\KPConv\Testing_classification.py", line 449, in
main()
File "c:\Users\aist2345\Documents\KPConv\Testing_classification.py", line 173, in main
classifier = load_checkpoint_with_fixes(classifier, train_model_path)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "c:\Users\aist2345\Documents\KPConv\Testing_classification.py", line 150, in load_checkpoint_with_fixes
missing_keys, unexpected_keys = model.load_state_dict(new_state_dict, strict=False)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "C:\Users\aist2345\AppData\Local\Packages\PythonSoftwareFoundation.Python.3.11_qbz5n2kfra8p0\LocalCache\local-packages\Python311\site-packages\torch\nn\modules\module.py", line 2189, in load_state_dict
raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format(
RuntimeError: Error(s) in loading state_dict for KPCNN:
size mismatch for head_mlp.mlp.weight: copying a param with shape torch.Size([1024, 1024]) from checkpoint, the shape in current model is torch.Size([1024, 64]).
size mismatch for head_softmax.mlp.weight: copying a param with shape torch.Size([40, 1024]) from checkpoint, the shape in current model is torch.Size([0, 1024]).
size mismatch for head_softmax.batch_norm.bias: copying a param with shape torch.Size([40]) from checkpoint, the shape in current model is torch.Size([0]).