Skip to content

GitLab

"git@developer.sourcefind.cn:norm/vllm.git" did not exist on "05bdf4eaf3bd8c577d09a6556acc3688094d0f6b"
Commit 1596bb28 authored by Yeqing Li's avatar Yeqing Li Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 356578560
parent 986b8776
Hide whitespace changes
Inline Side-by-side
Showing with 6 additions and 3 deletions
official/vision/beta/tasks/image_classification.py
View file @ 1596bb28
...@@ -68,7 +68,8 @@ class ImageClassificationTask(base_task.Task): ...@@ -68,7 +68,8 @@ class ImageClassificationTask(base_task.Task):
status = ckpt.restore(ckpt_dir_or_file) status = ckpt.restore(ckpt_dir_or_file)
status.expect_partial().assert_existing_objects_matched() status.expect_partial().assert_existing_objects_matched()
else: else:
assert "Only 'all' or 'backbone' can be used to initialize the model." raise ValueError(
"Only 'all' or 'backbone' can be used to initialize the model.")
logging.info('Finished loading pretrained checkpoint from %s', logging.info('Finished loading pretrained checkpoint from %s',
ckpt_dir_or_file) ckpt_dir_or_file)
......
official/vision/beta/tasks/maskrcnn.py
View file @ 1596bb28
...@@ -100,7 +100,8 @@ class MaskRCNNTask(base_task.Task): ...@@ -100,7 +100,8 @@ class MaskRCNNTask(base_task.Task):
status = ckpt.restore(ckpt_dir_or_file) status = ckpt.restore(ckpt_dir_or_file)
status.expect_partial().assert_existing_objects_matched() status.expect_partial().assert_existing_objects_matched()
else: else:
assert "Only 'all' or 'backbone' can be used to initialize the model." raise ValueError(
"Only 'all' or 'backbone' can be used to initialize the model.")
logging.info('Finished loading pretrained checkpoint from %s', logging.info('Finished loading pretrained checkpoint from %s',
ckpt_dir_or_file) ckpt_dir_or_file)
......
official/vision/beta/tasks/retinanet.py
View file @ 1596bb28
...@@ -78,7 +78,8 @@ class RetinaNetTask(base_task.Task): ...@@ -78,7 +78,8 @@ class RetinaNetTask(base_task.Task):
status = ckpt.restore(ckpt_dir_or_file) status = ckpt.restore(ckpt_dir_or_file)
status.expect_partial().assert_existing_objects_matched() status.expect_partial().assert_existing_objects_matched()
else: else:
assert "Only 'all' or 'backbone' can be used to initialize the model." raise ValueError(
"Only 'all' or 'backbone' can be used to initialize the model.")
logging.info('Finished loading pretrained checkpoint from %s', logging.info('Finished loading pretrained checkpoint from %s',
ckpt_dir_or_file) ckpt_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