dimension error in server.py

[Omenranr]

Hello, in server.py I get a dimension error: While copying the parameter named consensus.fc_fusion_scales.0.3.weight, whose dimensions in the model are torch.Size([1515, 256]) and whose dimensions in the checkpoint are torch.Size([27, 256]).

precisely in this line: net.load_state_dict(base_dict)