Fix int64_t error is ps mode

pull/8973/head
ZPaC 4 years ago
parent 661b6073a4
commit 8f6ba323cb

@ -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