How to add additional layer( inside the features module) in pre-trained EfficientNetB0 model? #343

Activity

1. sashika-himali

   changed the title [-]How to add additional layer in pre-trained EfficientNetB0 model?[/-] [+]How to add additional layer( inside the features module) in pre-trained EfficientNetB0 model?[/+] on Aug 6, 2023
2. cthugha10 commented on Dec 10, 2023

   

   cthugha10

   on Dec 10, 2023

   Last edited by cthugha10

   More actions

   Try just create another nn.mudule which includes the pre-trained model and your layer?
3. adhilcodes commented on Feb 16, 2024

   

   adhilcodes

   on Feb 16, 2024

   More actions

   class EfficientNet_bo(nn.Module):
       def __init__(self, num_classes = 2):
           super().__init__()
           self.model = EfficientNet.from_pretrained('efficientnet-b0')
   
           for param in self.model.parameters():
               param.requires_grad = False
   
           # Extract number of features from last layer
           in_features = self.model._fc.in_features
   
           self.model._fc = nn.Sequential(
               nn.Linear(in_features, 512),
               nn.ReLU(inplace=True),
               nn.Linear(512, 256),
               nn.ReLU(inplace=True),
               nn.Linear(256, num_classes),
               nn.Sigmoid() 
           )
   
       def forward(self, x):
           return  self.model(x)
       
   
   
   
   
       Here is a sample. Customise in the way you want it