size mismatch

[nvlabs]

size mismatch

I try to use Market, MSMT and cuhk03 dataset to train this model, so 2559 is the sum of these training kinds.

[layumi]

Hi @869250498
751 is the number of training classes in the Market dataset.
You need to modify the number for MSMT and CUHK03 in the config file.
https://github.com/NVlabs/DG-Net/blob/master/configs/latest.yaml#L6

Thank you for your reply @layumi
I have done that. But it doesn't work. I have used 'print' to find out that the ID_class of the config of ‘DGNet_Trainer’ is 2559. I hold the view that this error is caused from mismatch of network and input classes.

[layumi]

Have you re-trained the teacher model? The default teacher model is for Market-1501 (which contains 751 classes)

Yes I want to re-trained this model. Or is there any other train way? By the project Person_reID_baseline_pytorch?

[layumi]

You could use the baseline to train a basic model, and then use it as the new teacher model.

Thank you for your reply. It works