diff --git a/paddle/fluid/framework/downpour_worker.cc b/paddle/fluid/framework/downpour_worker.cc index 37e0c6ef22bc8e2e0ae5060c41e7aa13fd9ae7b4..dab3113fbc4e9eddc00704725175d7c4060a1287 100644 --- a/paddle/fluid/framework/downpour_worker.cc +++ b/paddle/fluid/framework/downpour_worker.cc @@ -122,7 +122,7 @@ void DownpourWorker::FillSparseValue(size_t table_idx) { auto& fea_value = feature_values_[table_id]; auto fea_idx = 0u; - std::vector init_value(table.emb_dim()); + std::vector init_value(table.fea_dim()); for (size_t i = 0; i < sparse_key_names_[table_id].size(); ++i) { std::string slot_name = sparse_key_names_[table_id][i]; std::string emb_slot_name = sparse_value_names_[table_id][i];