Skip to content

GitLab

"docs/git@developer.sourcefind.cn:OpenDAS/torch-scatter.git" did not exist on "6db8aad4ad78abab5d83099b5aefcd1f56631f77"
Commit ff2bf0b0 authored by Kaushik Shivakumar's avatar Kaushik Shivakumar
Browse files

add side inputs

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