add backticks to docstring string code keywords (#20863)

* add backticks to docstring string code keywords * Update remat.py
keras-team · Feb 6, 2025 · 43f40d8 · 43f40d8
1 parent c04cf9d
commit 43f40d8
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/keras/src/backend/common/remat.py b/keras/src/backend/common/remat.py
@@ -20,16 +20,16 @@ class RematScope:
     Args:
         mode: Rematerialization mode to apply.
             Options:
-            - "full": Apply rematerialization globally to all supported
+            - `"full"`: Apply rematerialization globally to all supported
               operations.
-            - "activations": Apply rematerialization to activations on any
+            - `"activations"`: Apply rematerialization to activations on any
               layers that contain `keras.activations` (e.g., `Dense(...,
               activation=relu)`).
-            - "larger_than": Apply rematerialization to layers with output sizes
-              larger than `output_size_threshold`.
-            - "list_of_layers": Apply rematerialization to a specific list of
+            - `"larger_than"`: Apply rematerialization to layers with output
+              sizes larger than `output_size_threshold`.
+            - `"list_of_layers"`: Apply rematerialization to a specific list of
               layer names.
-            - None: Disable rematerialization.
+            - `None`: Disable rematerialization.
         output_size_threshold: Output size threshold for the
             `"larger_than"` mode. Layers producing outputs larger than this
             threshold will be rematerialized. Default is `1024`.