Browse Source

fixed update embedding

tags/v1.2.0-rc1
chendongsheng 5 years ago
parent
commit
53ff0d63c2
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/ccsrc/ps/worker.cc

+ 1
- 1
mindspore/ccsrc/ps/worker.cc View File

@@ -383,7 +383,7 @@ void Worker::UpdateEmbeddingTable(const std::vector<Key> &keys, const std::vecto
sizes.push_back(kv_data.length());
}
}
worker_node_.Send(core::NodeRole::SERVER, rank_ids, data, sizes, 0);
worker_node_.Send(core::NodeRole::SERVER, rank_ids, data, sizes, kUpdateEmbeddingsCmd);
}

void Worker::Finalize() {


Loading…
Cancel
Save