Fix SmoothL1Loss op sigma attr problem.

pull/7391/head
liangchenghui 4 years ago
parent fa37f8fde3
commit 228c361903

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

Loading…
Cancel
Save