Skip to content

GitLab

"...resnet50_tensorflow.git" did not exist on "bf7483701da3954e74c8699d863b67c830b4e0de"
Unverified Commit 22864115 authored by srihari-humbarwadi's avatar srihari-humbarwadi
Browse files

set default checkpoint_path

parent 90156e0d
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
official/vision/beta/projects/panoptic_maskrcnn/tasks/panoptic_maskrcnn.py
View file @ 22864115
...@@ -66,6 +66,7 @@ class PanopticMaskRCNNTask(maskrcnn.MaskRCNNTask): ...@@ -66,6 +66,7 @@ class PanopticMaskRCNNTask(maskrcnn.MaskRCNNTask):
return return
def _get_checkpoint_path(checkpoint_dir_or_file): def _get_checkpoint_path(checkpoint_dir_or_file):
checkpoint_path = checkpoint_dir_or_file
if tf.io.gfile.isdir(checkpoint_dir_or_file): if tf.io.gfile.isdir(checkpoint_dir_or_file):
checkpoint_path = tf.train.latest_checkpoint( checkpoint_path = tf.train.latest_checkpoint(
checkpoint_dir_or_file) checkpoint_dir_or_file)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment