add workaround (intel#800)

delock · Apr 4, 2023 · cb355f0 · cb355f0
1 parent dabf27f
commit cb355f0
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/intel_extension_for_transformers/optimization/optimizer_tf.py b/intel_extension_for_transformers/optimization/optimizer_tf.py
@@ -880,6 +880,7 @@ def on_epoch_end(self, epoch, logs=None):
                 prune_model.save_pretrained(model_path, saved_model=True)
                 component.model = os.path.join(model_path, "saved_model/1")
                 component.model.model_type = "saved_model"
+                component.model.sess
                 hooks['on_epoch_end']()
 
             # pylint: disable=E1121