Skip to content

GitLab

"docs/git@developer.sourcefind.cn:change/sglang.git" did not exist on "cf5d27e30a7c2f74842347e994145f0ee2e9bbc1"
Commit a6797285 authored by Kaushik Shivakumar's avatar Kaushik Shivakumar
Browse files

fix

parent 3564e7ca
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
research/object_detection/model_lib_v2.py
View file @ a6797285
...@@ -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)
...@@ -604,7 +604,7 @@ def train_loop( ...@@ -604,7 +604,7 @@ def train_loop(
return strategy.reduce(tf.distribute.ReduceOp.SUM, return strategy.reduce(tf.distribute.ReduceOp.SUM,
per_replica_losses, axis=None) per_replica_losses, axis=None)
#@tf.function @tf.function
def _dist_train_step(data_iterator): def _dist_train_step(data_iterator):
"""A distributed train step.""" """A distributed train step."""
...@@ -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