Skip to content

GitLab

Commit 8948ba3f authored by Kaushik Shivakumar's avatar Kaushik Shivakumar
Browse files

fix file

parent a6797285
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
research/object_detection/model_lib_v2.py
View file @ 8948ba3f
...@@ -341,7 +341,7 @@ def load_fine_tune_checkpoint( ...@@ -341,7 +341,7 @@ 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)
...@@ -716,7 +716,7 @@ def eager_eval_loop( ...@@ -716,7 +716,7 @@ def eager_eval_loop(
evaluators = None evaluators = None
loss_metrics = {} loss_metrics = {}
#@tf.function @tf.function
def compute_eval_dict(features, labels): def compute_eval_dict(features, labels):
"""Compute the evaluation result on an image.""" """Compute the evaluation result on an image."""
# For evaling on train data, it is necessary to check whether groundtruth # For evaling on train data, it is necessary to check whether groundtruth
......
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