set default checkpoint_path

22864115 · srihari-humbarwadi · 90156e0d · 22864115
Unverified Commit 22864115 authored Aug 25, 2021 by srihari-humbarwadi
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

official/vision/beta/projects/panoptic_maskrcnn/tasks/panoptic_maskrcnn.py ...eta/projects/panoptic_maskrcnn/tasks/panoptic_maskrcnn.py +1 -0

No files found.
--- a/official/vision/beta/projects/panoptic_maskrcnn/tasks/panoptic_maskrcnn.py
+++ b/official/vision/beta/projects/panoptic_maskrcnn/tasks/panoptic_maskrcnn.py
@@ -66,6 +66,7 @@ class PanopticMaskRCNNTask(maskrcnn.MaskRCNNTask):
      return

    def _get_checkpoint_path(checkpoint_dir_or_file):
+      checkpoint_path = checkpoint_dir_or_file
      if tf.io.gfile.isdir(checkpoint_dir_or_file):
        checkpoint_path = tf.train.latest_checkpoint(
            checkpoint_dir_or_file)