Skip to content

GitLab

Commit 40ea6deb authored by Kaushik Shivakumar's avatar Kaushik Shivakumar
Browse files

fix

parent e6b7e0ad
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
research/object_detection/model_lib_v2.py
View file @ 40ea6deb
...@@ -120,7 +120,7 @@ def _compute_losses_and_predictions_dicts( ...@@ -120,7 +120,7 @@ def _compute_losses_and_predictions_dicts(
prediction_dict = model.predict( prediction_dict = model.predict(
preprocessed_images, preprocessed_images,
features[fields.InputDataFields.true_image_shape], features[fields.InputDataFields.true_image_shape],
model.get_side_inputs(features)) **model.get_side_inputs(features))
prediction_dict = ops.bfloat16_to_float32_nested(prediction_dict) prediction_dict = ops.bfloat16_to_float32_nested(prediction_dict)
losses_dict = model.loss( losses_dict = model.loss(
......
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