pytorch-multi-label-classifier icon indicating copy to clipboard operation
pytorch-multi-label-classifier copied to clipboard

questions about code

Open vickersmith opened this issue 6 years ago • 1 comments

your multi_label_classifier was constructed by several concurrent FC layers .Another idea is that the label is encoded by one-hot encoder and add a sigmoid function after the 3rd FC layer.At last ,only a CrossEntropyLoss is used without any concurrent FC layers.Do you think the idea works or have you try it?Thanks very much!

vickersmith avatar Oct 23 '18 12:10 vickersmith

I have't tried it yet. In fact, the weights of single fc layer proposed in your idea and the weigths of multiple concurrent fc layers have the same parameters number as long as the size of feature map before fc is identical.

pangwong avatar Oct 29 '18 08:10 pangwong