Skip to content

GitLab

Commit ec50c261 authored by Kaushik Shivakumar's avatar Kaushik Shivakumar
Browse files

fix model lib

parent 9da12cdd
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
research/object_detection/model_lib_v2.py
View file @ ec50c261
...@@ -111,7 +111,8 @@ def _compute_losses_and_predictions_dicts( ...@@ -111,7 +111,8 @@ 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))
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