未验证 提交 8f36c395 编写于 作者: C Chengmo 提交者: GitHub

Fix GEO-SGD init & send Bug (#22375)

* test=develop, fix geo Send & Init
上级 39bdefd8
...@@ -433,6 +433,7 @@ void GeoSgdCommunicator::Send(const std::vector<std::string> &sparse_var_names, ...@@ -433,6 +433,7 @@ void GeoSgdCommunicator::Send(const std::vector<std::string> &sparse_var_names,
} }
GeoSgdDenseParamInit(training_scope_, old_scope_.get(), local_var_name); GeoSgdDenseParamInit(training_scope_, old_scope_.get(), local_var_name);
} }
return;
} }
std::shared_ptr<SparseIdsMap> ids_table = std::make_shared<SparseIdsMap>(); std::shared_ptr<SparseIdsMap> ids_table = std::make_shared<SparseIdsMap>();
......
...@@ -61,7 +61,7 @@ class Communicator(object): ...@@ -61,7 +61,7 @@ class Communicator(object):
varnames = "&".join(vs["var_names"]) varnames = "&".join(vs["var_names"])
sections = "&".join([str(v) for v in vs["sections"]]) sections = "&".join([str(v) for v in vs["sections"]])
endpoints = "&".join(vs["epmap"]) endpoints = "&".join(vs["epmap"])
is_sparse = "1" if vs["is_sparse"] else "0" is_sparse = "1" if vs["is_sparse"] == ['True'] else "0"
push_var_names.append(k) push_var_names.append(k)
envs[k] = "#".join([varnames, sections, endpoints, is_sparse]) envs[k] = "#".join([varnames, sections, endpoints, is_sparse])
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册