From 5acf10941d59a069746c804399636e850a56ca62 Mon Sep 17 00:00:00 2001 From: peixu_ren Date: Wed, 19 Aug 2020 15:26:43 -0400 Subject: [PATCH] Check number validation of power in powertransform --- mindspore/nn/probability/bijector/power_transform.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/mindspore/nn/probability/bijector/power_transform.py b/mindspore/nn/probability/bijector/power_transform.py index 456f635818..1d9d2c5a88 100644 --- a/mindspore/nn/probability/bijector/power_transform.py +++ b/mindspore/nn/probability/bijector/power_transform.py @@ -15,6 +15,7 @@ """Power Bijector""" from mindspore.ops import operations as P from mindspore._checkparam import Validator as validator +from mindspore._checkparam import Rel from .bijector import Bijector class PowerTransform(Bijector): @@ -53,6 +54,7 @@ class PowerTransform(Bijector): param = dict(locals()) if param is None else param super(PowerTransform, self).__init__(name=name, param=param) validator.check_value_type('power', power, [int, float], self.name) + validator.check_number("power", power, 0, Rel.GE, self.name) self._power = power self.pow = P.Pow() self.exp = P.Exp()