rm: remove base model as unnecessary. adjust the Encoder class accordingly

Author: okunator

cellseg_models_pytorch/encoders/encoder.py

@@ -62,3 +62,8 @@ def forward(self, x: torch.Tensor) -> Tuple[torch.Tensor, Tuple[torch.Tensor, ..
         """Forward pass of the encoder and return all the features."""
         output, feats = self.encoder(x)
         return output, feats[::-1]
+
+    def freeze_encoder(self) -> None:
+        """Freeze the parameters of the encoeder."""
+        for param in self.encoder.parameters():
+            param.requires_grad = False