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

* test=develop, fix geo Send & Init
revert-22710-feature/integrated_ps_api
Chengmo 5 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);
}
return;
}
std::shared_ptr<SparseIdsMap> ids_table = std::make_shared<SparseIdsMap>();

@ -61,7 +61,7 @@ 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"
is_sparse = "1" if vs["is_sparse"] == ['True'] else "0"
push_var_names.append(k)
envs[k] = "#".join([varnames, sections, endpoints, is_sparse])

Loading…
Cancel
Save