From 2c697f834c1114b9fd39549835fb6d2fd5d47723 Mon Sep 17 00:00:00 2001 From: Zhan Lu <51200935+lausannel@users.noreply.github.com> Date: Fri, 17 Nov 2023 17:32:26 +0800 Subject: [PATCH] fix: typo in kafka broker --- HugeCTR/include/pybind/solver_wrapper.hpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/HugeCTR/include/pybind/solver_wrapper.hpp b/HugeCTR/include/pybind/solver_wrapper.hpp index 55ccc1001b..0bd767510b 100644 --- a/HugeCTR/include/pybind/solver_wrapper.hpp +++ b/HugeCTR/include/pybind/solver_wrapper.hpp @@ -146,7 +146,7 @@ void SolverPybind(pybind11::module& m) { pybind11::arg("all_reduce_algo") = AllReduceAlgo::NCCL, pybind11::arg("grouped_all_reduce") = false, pybind11::arg("num_iterations_statistics") = 20, pybind11::arg("perf_logging") = false, - pybind11::arg("drop_incomplete_batch") = true, pybind11::arg("kafka_brockers") = "", + pybind11::arg("drop_incomplete_batch") = true, pybind11::arg("kafka_brokers") = "", pybind11::arg("training_callbacks") = std::vector>()); }