Browse Source

Fix int64_t error is ps mode

tags/v1.1.0
ZPaC 5 years ago
parent
commit
8f6ba323cb
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      mindspore/ccsrc/ps/optimizer_info.cc

+ 2
- 2
mindspore/ccsrc/ps/optimizer_info.cc View File

@@ -296,7 +296,7 @@ SparseAdamOptimInfo::SparseAdamOptimInfo(const AddressPtr &weight, const Address
inputs_.push_back(grad);
inputs_.push_back(indices);
grads_offset_ = grad->size / sizeof(float);
indices_offset_ = indices->size / sizeof(int64_t);
indices_offset_ = indices->size / sizeof(int);
}

void SparseAdamOptimInfo::Update(const Values &values, const Lengths &lens) {
@@ -340,7 +340,7 @@ SparseFtrlOptimInfo::SparseFtrlOptimInfo(const AddressPtr &weight, const Address
inputs_.push_back(grad);
inputs_.push_back(indices);
grads_offset_ = grad->size / sizeof(float);
indices_offset_ = indices->size / sizeof(int64_t);
indices_offset_ = indices->size / sizeof(int);
}

const AddressPtr &SparseFtrlOptimInfo::gradient() {


Loading…
Cancel
Save