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

* test=develop, fix geo Send & Init
revert-22710-feature/integrated_ps_api
Chengmo 6 years ago committed by GitHub
parent 39bdefd875
commit 8f36c39537
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -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):
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])

Loading…
Cancel
Save