diff --git a/src/core/src/op/proposal.cpp b/src/core/src/op/proposal.cpp index 96d094303fc..18941d3e014 100644 --- a/src/core/src/op/proposal.cpp +++ b/src/core/src/op/proposal.cpp @@ -98,6 +98,7 @@ void op::v4::Proposal::validate_and_infer_types() { const auto& input0_type = get_input_element_type(0); set_output_type(0, input0_type, output_shapes[0]); set_output_type(1, input0_type, output_shapes[1]); + m_attrs.infer_probs = true; // Proposal v4 requires default true of infer_probs so output_1 has valid data. } std::shared_ptr op::v4::Proposal::clone_with_new_inputs(const OutputVector& new_args) const {