EfficientNet-PyTorch
EfficientNet-PyTorch copied to clipboard
multi-head efficientNet
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!