!7939 add valid check

Merge pull request !7939 from lijiaqi/raise
pull/7939/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
commit f187f8211f

@ -316,7 +316,8 @@ class Parameter(MetaTensor_):
Args:
data (Union[Tensor, MetaTensor, int, float]): new data.
slice_shape (bool): If slice the Parameter, will not check if shape is match. Default: False.
slice_shape (bool): If slice the parameter is set to true, the shape is not checked for consistency.
Default: False.
Returns:
Parameter, the parameter after set data.

@ -336,6 +336,9 @@ class Adam(Optimizer):
def target(self, value):
"""If the input value is set to "CPU", the parameters will be updated on the host using the Fused
optimizer operation."""
if not isinstance(value, str):
raise ValueError("The value must be str type, but got value type is {}".format(type(value)))
if value not in ('CPU', 'Ascend'):
raise ValueError("The value must be 'CPU' or 'Ascend', but got value {}".format(value))

@ -189,6 +189,8 @@ class FTRL(Optimizer):
def target(self, value):
"""If the input value is set to "CPU", the parameters will be updated on the host using the Fused
optimizer operation."""
if not isinstance(value, str):
raise ValueError("The value must be str type, but got value type is {}".format(type(value)))
if value not in ('CPU', 'Ascend'):
raise ValueError("The value must be 'CPU' or 'Ascend', but got value {}".format(value))

@ -254,6 +254,9 @@ class LazyAdam(Optimizer):
def target(self, value):
"""If the input value is set to "CPU", the parameters will be updated on the host using the Fused
optimizer operation."""
if not isinstance(value, str):
raise ValueError("The value must be str type, but got value type is {}".format(type(value)))
if value not in ('CPU', 'Ascend'):
raise ValueError("The value must be 'CPU' or 'Ascend', but got value {}".format(value))

@ -158,6 +158,8 @@ class ProximalAdagrad(Optimizer):
def target(self, value):
"""If the input value is set to "CPU", the parameters will be updated on the host using the Fused
optimizer operation."""
if not isinstance(value, str):
raise ValueError("The value must be str type, but got value type is {}".format(type(value)))
if value not in ('CPU', 'Ascend'):
raise ValueError("The value must be 'CPU' or 'Ascend', but got value {}".format(value))

Loading…
Cancel
Save