From 43f40d81b2d444bb89d9513c8d78032209b4fa5c Mon Sep 17 00:00:00 2001 From: Divyashree Sreepathihalli Date: Thu, 6 Feb 2025 10:37:22 -0800 Subject: [PATCH] add backticks to docstring string code keywords (#20863) * add backticks to docstring string code keywords * Update remat.py --- keras/src/backend/common/remat.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/keras/src/backend/common/remat.py b/keras/src/backend/common/remat.py index cfad2bf8d62..430079c17af 100644 --- 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`.