`class VisionTransformer(nn.Module): 
 def forward(self, x):     
       x = self.embedding_layer(x)  
       x = self.transformer(x) 
       x = self.post_transformer_ln(x)         
      x = self.cls_layer(x)   
    return x`

i need to extract the features without using cls_layer i tried to remove it but got
ValueError: Cannot create stem without list of sizes
which is depend on another class… how can i control the model to freeze the layer?