RuntimeError: Error(s) in loading state_dict for Model: size mismatch for tree_model.0.weight: copying a param with shape torch.Size([31, 2048]) from checkpoint, the shape in current model is torch.Size([15, 2048]).

MichalZnalezniak / Contrastive-Hierarchical-Clustering

This is the official code for the paper Contrastive Hierarchical Clustering (ECML PKDD 2023)

15 stars 3 forks source link

RuntimeError: Error(s) in loading state_dict for Model: size mismatch for tree_model.0.weight: copying a param with shape torch.Size([31, 2048]) from checkpoint, the shape in current model is torch.Size([15, 2048]). #7

Closed Vanguitar closed 9 months ago

Vanguitar commented 9 months ago

Dear author,

There is an error when I evaluate the dataset cifar10. I run the command: python evaluate.py --save_point ./pre-trained/CIFAR10/models/ --dataset-name cifar10 # Generate metrics for CIFAR10 and then the error occurs:

Vanguitar commented 9 months ago

It seems the system has a moment. The code is correct.