Update reproducibility_recipes.py

Frightera · web-flow · commit 6bf16bf4046c · 2024-10-02T01:11:27.000+03:00
diff --git a/examples/keras_recipes/reproducibility_recipes.py b/examples/keras_recipes/reproducibility_recipes.py
@@ -151,8 +151,10 @@ def train_model(train_data: tf.data.Dataset, test_data: tf.data.Dataset) -> dict
     )
 
     model.compile(
-        optimizer="adam", loss="sparse_categorical_crossentropy", metrics=["accuracy"]
+        optimizer="adam", loss="sparse_categorical_crossentropy", metrics=["accuracy"], jit_compile = False
     )
+    # jit_compile's default value is "auto" which will cause some problems in some
+    # ops, therefore it's set to False. 
 
     # model.fit has a `shuffle` parameter which has a default value of `True`.
     # If you are using array-like objects, this will shuffle the data before

Original file line number	Diff line number	Diff line change
`@@ -151,8 +151,10 @@ def train_model(train_data: tf.data.Dataset, test_data: tf.data.Dataset) -> dict`
`151`	`151`	`)`
`152`	`152`
`153`	`153`	`model.compile(`
`154`		`- optimizer="adam", loss="sparse_categorical_crossentropy", metrics=["accuracy"]`
	`154`	`+ optimizer="adam", loss="sparse_categorical_crossentropy", metrics=["accuracy"], jit_compile = False`
`155`	`155`	`)`
	`156`	`+ # jit_compile's default value is "auto" which will cause some problems in some`
	`157`	`+ # ops, therefore it's set to False.`
`156`	`158`
`157`	`159`	# model.fit has a `shuffle` parameter which has a default value of `True`.
`158`	`160`	`# If you are using array-like objects, this will shuffle the data before`