EfficientNet-PyTorch icon indicating copy to clipboard operation
EfficientNet-PyTorch copied to clipboard

multi-head efficientNet

Open www7890 opened this issue 4 years ago • 0 comments

Hi all, I want to add a new head for a simple two classes classification task, I will have 2 , 6 classes to classify. How do I add a new head and only has one body of model?

Thanks!

www7890 avatar Aug 13 '21 03:08 www7890