Train on custom dataset (with more no. of classes)

[ankurguria]

@lannguyen0910
How can I train the model on my custom dataset (number of classes: 2504) using the pre-train weights (number of classes: 751 for market).
I came across a method of removing the last or fully connected layer before loading the weights and then adding the fully connected layer (with desired num_classes) but wasn't able to implement it.
Could you please help in implementing the same?

Thanks and regards.