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

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