Skip to content

Commit

Permalink
Fix for pre-commit.ci
Browse files Browse the repository at this point in the history
  • Loading branch information
zldrobit authored and glenn-jocher committed Feb 4, 2022
1 parent ae92803 commit 8c231ef
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion export.py
Original file line number Diff line number Diff line change
Expand Up @@ -263,7 +263,7 @@ def export_saved_model(model, im, file, dynamic,
tfm,
f,
options=tf.saved_model.SaveOptions() if parse_version(tf.__version__) < parse_version("2.6") else
tf.saved_model.SaveOptions(experimental_custom_gradients=False))
tf.saved_model.SaveOptions(experimental_custom_gradients=False))
LOGGER.info(f'{prefix} export success, saved as {f} ({file_size(f):.1f} MB)')
return keras_model, f
except Exception as e:
Expand Down

0 comments on commit 8c231ef

Please sign in to comment.