Update save_pruned_model.py

lucas-ecm · Feb 11, 2023 · 4408024 · 4408024
1 parent 66d17ae
commit 4408024
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/save_pruned_model.py b/save_pruned_model.py
@@ -99,7 +99,7 @@ def apply_pruning_to_dense(layer):
   # Use `tf.keras.models.clone_model` to apply `apply_pruning_to_dense` 
   # to the layers of the model.
   model_for_pruning = tf.keras.models.clone_model(
-      base_model,
+      model,
       clone_function=apply_pruning_to_dense,
   )