Update save_pruned_model.py

lucas-ecm · Feb 25, 2023 · 941ecd9 · 941ecd9
1 parent aff9539
commit 941ecd9
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/save_pruned_model.py b/save_pruned_model.py
@@ -23,6 +23,7 @@ def get_gzipped_model_size(model):
   # Returns size of gzipped model, in bytes.
   import os
   import zipfile
+  import tempfile
 
   _, keras_file = tempfile.mkstemp('.h5')
   model.save(keras_file, include_optimizer=False)