forked from mindspore-Ecosystem/mindspore
add valid check
This commit is contained in:
parent
fff8dff071
commit
c4ace78d1c
|
@ -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…
Reference in New Issue