Hi all,
I ran the transfer learning on my datasets, achieved n classes on the top of the network, but my purpose is simply expanding the categories of a pretrained model (such as resnet18 which trained on ImageNet and can predict thousand categories) and add a new category. The advantage of this approach is avoiding a situation in which the input is X1 (categorized by the pretrained categories) and my network mistakenly predict one of my trained classes (of the n classes that I trained).