Hy guys, i would use transfer learning with resnet50. I use the KLDivLoss for the divergence of 2 distribution. Is there a possibility to change FC and have an output dimension of [9,224,224] (the dimension of a photo with 9 channels(my second distribution)?