search

Raschka-research-group / coral-cnn

Rank Consistent Ordinal Regression for Neural Networks with Application to Age Estimation
https://www.sciencedirect.com/science/article/pii/S016786552030413X
MIT License
335 stars 62 forks source link

The question about fc layer #21

Closed mengxingkong closed 4 years ago

mengxingkong commented 4 years ago

self.fc = nn.Linear(2048 * block.expansion, 1, bias=False) why the output of fc layer is 1 other than num_classes? logits = logits + self.linear_1_bias one output feature of the fc layer plus num_classes bias?

rasbt commented 4 years ago

That's because of the weight sharing explained in the paper. Let me know if that's unclear.