@@ -44,7 +44,7 @@ def __init__(
44
44
nrow : int = 8 ,
45
45
padding : int = 2 ,
46
46
normalize : bool = True ,
47
- norm_range : Optional [Tuple [int , int ]] = None ,
47
+ value_range : Optional [Tuple [int , int ]] = None ,
48
48
scale_each : bool = False ,
49
49
pad_value : int = 0 ,
50
50
) -> None :
@@ -56,7 +56,7 @@ def __init__(
56
56
padding: Amount of padding. Default: ``2``.
57
57
normalize: If ``True``, shift the image to the range (0, 1),
58
58
by the min and max values specified by :attr:`range`. Default: ``False``.
59
- norm_range : Tuple (min, max) where min and max are numbers,
59
+ value_range : Tuple (min, max) where min and max are numbers,
60
60
then these numbers are used to normalize the image. By default, min and max
61
61
are computed from the tensor.
62
62
scale_each: If ``True``, scale each image in the batch of
@@ -71,7 +71,7 @@ def __init__(
71
71
self .nrow = nrow
72
72
self .padding = padding
73
73
self .normalize = normalize
74
- self .norm_range = norm_range
74
+ self .value_range = value_range
75
75
self .scale_each = scale_each
76
76
self .pad_value = pad_value
77
77
@@ -81,7 +81,7 @@ def _to_grid(self, images):
81
81
nrow = self .nrow ,
82
82
padding = self .padding ,
83
83
normalize = self .normalize ,
84
- value_range = self .norm_range ,
84
+ value_range = self .value_range ,
85
85
scale_each = self .scale_each ,
86
86
pad_value = self .pad_value ,
87
87
)
0 commit comments