Browse Source

modify spell error

tags/v0.3.0-alpha
chang zherui 6 years ago
parent
commit
b57b111947
1 changed files with 1 additions and 2 deletions
  1. +1
    -2
      mindspore/ops/operations/nn_ops.py

+ 1
- 2
mindspore/ops/operations/nn_ops.py View File

@@ -2189,7 +2189,7 @@ class SparseApplyFtrlD(PrimitiveWithInfer):
self.l1 = validator.check_type("l1", l1, [float])
self.l2 = validator.check_type("l2", l2, [float])
self.lr_power = validator.check_type("lr_power", lr_power, [float])
self.use_locking = validator.check_type("use_locking", use_locaking, [bool])
self.use_locking = validator.check_type("use_locking", use_locking, [bool])

def infer_shape(self, var_shape, accum_shape, linear_shape, grad_shape, indices_shape):
validator.check_param_equal('var shape', var_shape, 'accum shape', accum_shape)
@@ -2210,7 +2210,6 @@ class SparseApplyFtrlD(PrimitiveWithInfer):
validator.check_subclass("linear_type", linear_type, mstype.tensor)
validator.check_subclass("grad_type", grad_type, mstype.tensor)
validator.check_subclass("indices_type", indices_type, mstype.tensor)
validator.check_subclass('indices_type', indices_type, [mstype.int32])

return var_type



Loading…
Cancel
Save