未验证 提交 13dca2c9 编写于 作者: C Chengmo 提交者: GitHub

[Cherry-pick]Fix geo init & send (#22413)

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

* test=develop, fix geo Send & Init

* test=release/1.7,test=develop, cherry-pick 8f36c395
上级 e87ddb28
......@@ -433,6 +433,7 @@ void GeoSgdCommunicator::Send(const std::vector<std::string> &sparse_var_names,
}
GeoSgdDenseParamInit(training_scope_, old_scope_.get(), local_var_name);
}
return;
}
std::shared_ptr<SparseIdsMap> ids_table = std::make_shared<SparseIdsMap>();
......
......@@ -61,7 +61,8 @@ class Communicator(object):
varnames = "&".join(vs["var_names"])
sections = "&".join([str(v) for v in vs["sections"]])
endpoints = "&".join(vs["epmap"])
is_sparse = "1" if vs["is_sparse"] else "0"
# record parameter sparse or dense
is_sparse = "1" if vs["is_sparse"] == ['True'] else "0"
push_var_names.append(k)
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.
先完成此消息的编辑!
想要评论请 注册