fix: minor style mods + default arg changes

Author: okunator

cellseg_models_pytorch/modules/mlp.py

@@ -12,7 +12,7 @@ class Mlp(nn.Module):
     def __init__(
         self,
         in_channels: int,
-        mlp_ratio: int = 4,
+        mlp_ratio: int = 2,
         activation: str = "star_relu",
         dropout: float = 0.0,
         bias: bool = False,
@@ -31,7 +31,7 @@ def __init__(
         ----------
             in_channels : int
                 Number of input features.
-            mlp_ratio : int, default=4
+            mlp_ratio : int, default=2
                 Scaling factor to get the number hidden features from the `in_features`.
             activation : str, default="star_relu"
                 The name of the activation function.
@@ -69,7 +69,7 @@ class MlpBlock(nn.Module):
     def __init__(
         self,
         in_channels: int,
-        mlp_ratio: int = 4,
+        mlp_ratio: int = 2,
         activation: str = "star_relu",
         activation_kwargs: Dict[str, Any] = None,
         dropout: float = 0.0,
@@ -85,7 +85,7 @@ def __init__(
         ----------
             in_channels : int
                 Number of input features.
-            mlp_ratio : int, default=4
+            mlp_ratio : int, default=2
                 Scaling factor to get the number hidden features from the `in_features`.
             activation : str, default="star_relu"
                 The name of the activation function.

cellseg_models_pytorch/modules/patch_embeddings.py

@@ -21,7 +21,7 @@ def __init__(
         flatten: bool = True,
         normalization: str = None,
         norm_kwargs: Dict[str, Any] = None,
-        **kwargs
+        **kwargs,
     ) -> None:
         """Patch an image with nn.Conv2d and then embed.
 
@@ -153,7 +153,7 @@ def __init__(
         head_dim: int = 64,
         num_heads: int = 8,
         normalization: int = None,
-        **norm_kwargs
+        **norm_kwargs,
     ) -> None:
         """Patch an input image and then embed/project.