Generate nb and md files

nkovela1 · Aug 4, 2023 · 80d450a · 80d450a
1 parent 1e92e4f
commit 80d450a
Show file tree

Hide file tree

Showing 3 changed files with 837 additions and 9 deletions.
diff --git a/guides/customizing_saving_and_serialization.py b/guides/customizing_saving_and_serialization.py
@@ -4,6 +4,7 @@
 Date created: 2023/03/15
 Last modified: 2023/03/15
 Description: A more advanced guide on customizing saving for your layers and models.
+Accelerator: None
 """
 
 """
@@ -78,7 +79,9 @@ def save_own_variables(self, store):
 class LayerWithCustomVariables(keras.layers.Dense):
     def __init__(self, units, **kwargs):
         super().__init__(units, **kwargs)
-        self.stored_variables = tf.Variable(np.random.random((10,)), name="special_arr", dtype=tf.float32)
+        self.stored_variables = tf.Variable(
+            np.random.random((10,)), name="special_arr", dtype=tf.float32
+        )
 
     def save_own_variables(self, store):
         super().save_own_variables(store)
@@ -98,9 +101,7 @@ def call(self, inputs):
         return super().call(inputs) * self.stored_variables
 
 
-model = keras.Sequential([
-    LayerWithCustomVariables(1)
-])
+model = keras.Sequential([LayerWithCustomVariables(1)])
 
 ref_input = np.random.random((8, 10))
 ref_output = np.random.random((8,))
@@ -112,7 +113,7 @@ def call(self, inputs):
 
 np.testing.assert_allclose(
     model.layers[0].stored_variables.numpy(),
-    restored_model.layers[0].stored_variables.numpy()
+    restored_model.layers[0].stored_variables.numpy(),
 )
 
 """
@@ -149,9 +150,9 @@ def load_assets(self, inner_path):
         self.vocab = text.replace("<unk>", "little")
 
 
-model = keras.Sequential([
-    LayerWithCustomAssets(vocab='Mary had a <unk> lamb.', units=5)
-])
+model = keras.Sequential(
+    [LayerWithCustomAssets(vocab="Mary had a <unk> lamb.", units=5)]
+)
 
 x = np.random.random((10, 10))
 y = model(x)
@@ -338,4 +339,4 @@ def compile_from_config(self, config):
 - `get_compile_config` and `compile_from_config` save and restore the model's
 compiled states.
 
-"""
+"""