Skip to content

GitLab

Commit 7a02b5ce authored by Vighnesh Birodkar's avatar Vighnesh Birodkar Committed by TF Object Detection Team
Browse files

Restore dummy computation function. 

PiperOrigin-RevId: 364388006
parent 2da86542
Hide whitespace changes
Inline Side-by-side
Showing with 26 additions and 26 deletions
research/object_detection/model_lib_v2.py
View file @ 7a02b5ce
...@@ -351,32 +351,32 @@ def load_fine_tune_checkpoint( ...@@ -351,32 +351,32 @@ def load_fine_tune_checkpoint(
features, labels = iter(input_dataset).next() features, labels = iter(input_dataset).next()
# @tf.function @tf.function
# def _dummy_computation_fn(features, labels): def _dummy_computation_fn(features, labels):
# model._is_training = False # pylint: disable=protected-access model._is_training = False # pylint: disable=protected-access
# tf.keras.backend.set_learning_phase(False) tf.keras.backend.set_learning_phase(False)
# labels = model_lib.unstack_batch( labels = model_lib.unstack_batch(
# labels, unpad_groundtruth_tensors=unpad_groundtruth_tensors) labels, unpad_groundtruth_tensors=unpad_groundtruth_tensors)
# return _compute_losses_and_predictions_dicts( return _compute_losses_and_predictions_dicts(
# model, model,
# features, features,
# labels) labels)
# strategy = tf.compat.v2.distribute.get_strategy() strategy = tf.compat.v2.distribute.get_strategy()
# if hasattr(tf.distribute.Strategy, 'run'): if hasattr(tf.distribute.Strategy, 'run'):
# strategy.run( strategy.run(
# _dummy_computation_fn, args=( _dummy_computation_fn, args=(
# features, features,
# labels, labels,
# )) ))
# else: else:
# strategy.experimental_run_v2( strategy.experimental_run_v2(
# _dummy_computation_fn, args=( _dummy_computation_fn, args=(
# features, features,
# labels, labels,
# )) ))
restore_from_objects_dict = model.restore_from_objects( restore_from_objects_dict = model.restore_from_objects(
fine_tune_checkpoint_type=checkpoint_type) fine_tune_checkpoint_type=checkpoint_type)
......
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