Browse Source

!7576 modify ftrl and proximal primivite target

Merge pull request !7576 from lijiaqi/ftrl
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
ef3d05475f
2 changed files with 2 additions and 2 deletions
  1. +1
    -1
      mindspore/nn/optim/ftrl.py
  2. +1
    -1
      mindspore/nn/optim/proximal_ada_grad.py

+ 1
- 1
mindspore/nn/optim/ftrl.py View File

@@ -195,7 +195,7 @@ class FTRL(Optimizer):

if value == 'CPU':
self.sparse_opt = P.FusedSparseFtrl(self.lr, self.l1, self.l2, self.lr_power, self.use_locking)
self.sparse_opt.add_prim_attr("primitive", "CPU")
self.sparse_opt.add_prim_attr("primitive_target", "CPU")
else:
self.sparse_opt = P.SparseApplyFtrl(self.lr, self.l1, self.l2, self.lr_power, self.use_locking)



+ 1
- 1
mindspore/nn/optim/proximal_ada_grad.py View File

@@ -163,7 +163,7 @@ class ProximalAdagrad(Optimizer):
raise ValueError("The value must be 'CPU' or 'Ascend', but got value {}".format(value))
if value == 'CPU':
self.sparse_opt = P.FusedSparseProximalAdagrad(self.use_locking).add_prim_attr("primitive", "CPU")
self.sparse_opt = P.FusedSparseProximalAdagrad(self.use_locking).add_prim_attr("primitive_target", "CPU")
else:
self.sparse_opt = P.SparseApplyProximalAdagrad(self.use_locking)


Loading…
Cancel
Save