Browse Source

[MNT] modify error msg

tags/v0.3.2
shihy 2 years ago
parent
commit
9be1495a2f
1 changed files with 1 additions and 3 deletions
  1. +1
    -3
      learnware/specification/regular/image/rkme.py

+ 1
- 3
learnware/specification/regular/image/rkme.py View File

@@ -98,9 +98,7 @@ class RKMEImageSpecification(RegularStatsSpecification):

"""
if len(X.shape) != 4:
raise ValueError(
"X should be in shape of [N, C, {0:d}, {0:d}]. ".format(RKMEImageSpecification.IMAGE_WIDTH)
)
raise ValueError("X should be in shape of [N, C, H, W]. ")

if (
X.shape[2] != RKMEImageSpecification.IMAGE_WIDTH or X.shape[3] != RKMEImageSpecification.IMAGE_WIDTH


Loading…
Cancel
Save