ShuffleNet-Series icon indicating copy to clipboard operation
ShuffleNet-Series copied to clipboard

Low accuracy on multi-clsss problem

Open IzouGend opened this issue 3 years ago • 1 comments

When I used shufflenet to train models on my own dataset, I found shufflenet has high accuracy on two-class classification probelm while low accuracy on multi-class problem. I used finetuning based on officially released pretrained model. Is it unproper to use pretrained model when training multi-class dataset?

IzouGend avatar Apr 25 '21 02:04 IzouGend

@IzouGend It is ok to use the ImageNet pretrained model because ImageNet is a multi-class classification task.

nmaac avatar Apr 25 '21 03:04 nmaac