RuntimeError: Error(s) in loading state_dict for Net:

created at 11-21-2021 views: 5

description

size mismatch for classifier.4.weight: copying a param with shape torch.Size([7, 256]) from checkpoint, the shape in current model is torch.Size([751, 256]).
size mismatch for classifier.4.bias: copying a param with shape torch.Size([7]) from checkpoint, the shape in current model is torch.Size([751]).

solution

Train the tracking weights of deepsort, use the default data set market1501, replace it with your own data set, and perform the test with the trained weights and report the error as above. 

Change the num_class in model.py to the number of your own classes.  

For example, my num_class is 7, market1501 defaults to 751 class)

class Net(nn.Module):
     def __init__(self, num_classes=751, reid=False):#Change the number of training classes
created at:11-21-2021
edited at: 11-21-2021: