Browse Source

Fix SmoothL1Loss op sigma attr problem.

tags/v1.1.0
liangchenghui 5 years ago
parent
commit
228c361903
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      mindspore/ops/operations/nn_ops.py

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

@@ -1884,6 +1884,7 @@ class SmoothL1Loss(PrimitiveWithInfer):
validator.check_value_type('beta', beta, [float], self.name)
validator.check('beta', beta, '', 0, Rel.GT, self.name)
self.init_prim_io_names(inputs=['prediction', 'target'], outputs=['output'])
self.add_prim_attr('sigma', beta)

def infer_shape(self, prediction, target):
validator.check('prediction shape', prediction, 'target shape', target, Rel.EQ, self.name)


Loading…
Cancel
Save