bugfix

lucas-ecm · Feb 11, 2023 · 463fbe6 · 463fbe6
1 parent 10d0d0f
commit 463fbe6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/save_pruned_model.py b/save_pruned_model.py
@@ -89,7 +89,7 @@ def save_tf():
                                                                   final_sparsity=FLAGS.final_sparsity,
                                                                   begin_step=0,
                                                                   end_step=end_step)
-  #}
+  }
 
   def apply_pruning_to_dense(layer):
     if isinstance(layer, tf.keras.layers.Conv2D):