Add docstring to create_act_layer function

timm/layers/create_act.py

@@ -90,9 +90,16 @@
 
 
 def get_act_fn(name: Optional[LayerType] = 'relu'):
-    """ Activation Function Factory
-    Fetching activation fns by name with this function allows export or torch script friendly
-    functions to be returned dynamically based on current config.
+    """Fetch an activation function by name.
+
+    Returns export/script-friendly or memory-efficient activation functions
+    dynamically based on current config.
+
+    Args:
+        name: Activation function name, callable, or None.
+
+    Returns:
+        Activation function or None.
     """
     if not name:
         return None
@@ -108,9 +115,16 @@ def get_act_fn(name: Optional[LayerType] = 'relu'):
 
 
 def get_act_layer(name: Optional[LayerType] = 'relu'):
-    """ Activation Layer Factory
-    Fetching activation layers by name with this function allows export or torch script friendly
-    functions to be returned dynamically based on current config.
+    """Fetch an activation layer by name.
+
+    Returns export/script-friendly or memory-efficient activation layers
+    dynamically based on current config.
+
+    Args:
+        name: Activation layer name, type, callable, or None.
+
+    Returns:
+        Activation layer class or None.
     """
     if name is None:
         return None
@@ -131,6 +145,19 @@ def create_act_layer(
         inplace: Optional[bool] = None,
         **kwargs
 ):
+    """Create an activation layer instance by name.
+
+    Handles inplace argument for activations that support it, gracefully
+    falling back for those that don't.
+
+    Args:
+        name: Activation layer name or type.
+        inplace: Enable inplace operation if supported by the activation.
+        **kwargs: Additional arguments passed to activation layer.
+
+    Returns:
+        Instantiated activation layer or None.
+    """
     act_layer = get_act_layer(name)
     if act_layer is None:
         return None