Browse Source

!5774 Add data type check

Merge pull request !5774 from lilei/add_check
tags/v1.0.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
dbde51e615
1 changed files with 4 additions and 0 deletions
  1. +4
    -0
      mindspore/common/initializer.py

+ 4
- 0
mindspore/common/initializer.py View File

@@ -464,6 +464,10 @@ def initializer(init, shape=None, dtype=mstype.float32):
elif isinstance(shape, numbers.Number):
shape = (shape,)

for value in shape if shape is not None else ():
if not isinstance(value, int) or value <= 0:
raise ValueError(f"shape is invalid, shape value must be positive integer, shape:{shape}")

if isinstance(init, Initializer):
init.shape = init.shape if init.shape is not None else shape
init.dtype = init.dtype if init.dtype is not None else dtype


Loading…
Cancel
Save