From 762f8178aefa71da62a856ad579b83879d95d2e5 Mon Sep 17 00:00:00 2001
From: laipaang <laipaang@gmail.com>
Date: Fri, 2 Jul 2021 15:29:17 +0800
Subject: [PATCH] init_metric default arg

---
 paddle/fluid/pybind/box_helper_py.cc | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/paddle/fluid/pybind/box_helper_py.cc b/paddle/fluid/pybind/box_helper_py.cc
index a797c9c8bddeb..9969cc557276d 100644
--- a/paddle/fluid/pybind/box_helper_py.cc
+++ b/paddle/fluid/pybind/box_helper_py.cc
@@ -85,6 +85,19 @@ void BindBoxWrapper(py::module* m) {
       .def("initialize_auc_runner", &framework::BoxWrapper::InitializeAucRunner,
            py::call_guard<py::gil_scoped_release>())
       .def("init_metric", &framework::BoxWrapper::InitMetric,
+           py::arg("method"),
+           py::arg("name"),
+           py::arg("label_varname"),
+           py::arg("pred_varname"),
+           py::arg("cmatch_rank_varname"),
+           py::arg("mask_varname"),
+           py::arg("metric_phase"),
+           py::arg("cmatch_rank_group"),
+           py::arg("ignore_rank"),
+           py::arg("bucket_size") = 1000000,
+           py::arg("mode_collect_in_gpu") = false,
+           py::arg("max_batch_size") = 0,
+           py::arg("sample_scale_varnam") = "",
            py::call_guard<py::gil_scoped_release>())
       .def("get_metric_msg", &framework::BoxWrapper::GetMetricMsg,
            py::call_guard<py::gil_scoped_release>())