Internal change

PiperOrigin-RevId: 384957416

Internal change
PiperOrigin-RevId: 384957416
d095658a · Hongkun Yu · A. Unique TensorFlower · 2a185c1f · d095658a
Commit d095658a authored Jul 15, 2021 by Hongkun Yu Committed by A. Unique TensorFlower Jul 15, 2021
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

official/core/export_base.py official/core/export_base.py +1 -1

No files found.
--- a/official/core/export_base.py
+++ b/official/core/export_base.py
@@ -82,7 +82,7 @@ def export(export_module: ExportModule,
    The savedmodel directory path.
  """
  ckpt_dir_or_file = checkpoint_path
-  if tf.io.gfile.isdir(ckpt_dir_or_file):
+  if ckpt_dir_or_file is not None and tf.io.gfile.isdir(ckpt_dir_or_file):
    ckpt_dir_or_file = tf.train.latest_checkpoint(ckpt_dir_or_file)
  if ckpt_dir_or_file:
    checkpoint = tf.train.Checkpoint(model=export_module.model)