update comment

999b9ff5 · LielinJiang · 76876f52 · 999b9ff5
隐藏空白更改
内联并排

Showing with 3 addition and 1 deletion

transform/transforms.py transform/transforms.py +3 -1

未找到文件。
--- a/transform/transforms.py
+++ b/transform/transforms.py
@@ -297,7 +297,9 @@ class Permute(object):
    """
    def __init__(self, mode="CHW"):
-        assert mode in ["CHW"], "unsupported mode: {}".format(mode)
+        assert mode in [
+            "CHW"
+        ], "Only support 'CHW' mode, but received mode: {}".format(mode)
        self.mode = mode
    def __call__(self, img):