It seems the class number is used to build the model, so different class number between pretrained model and custom dataset can cause error?
I saw in train.py, the create_model function use "skip_mismatch", is this the way to solve the metioned problem?
It seems the class number is used to build the model, so different class number between pretrained model and custom dataset can cause error? I saw in train.py, the create_model function use "skip_mismatch", is this the way to solve the metioned problem?