ptrblck:

In that case you would have to reduce the output channels from 2048 to the number of classes, e.g. by using a conv layer with out_channels=nb_classes . The kernel size might be 1x1 , but it depends on your use case, if that’s the best approach.


sorry but i don’t think i fully understand.