We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 1b60fe6 commit f8d84edCopy full SHA for f8d84ed
torchvision/transforms/v2/_augment.py
@@ -77,13 +77,13 @@ def __init__(
77
self.scale = scale
78
self.ratio = ratio
79
if isinstance(value, (int, float)):
80
- self.value = [float(value)]
+ value = [float(value)]
81
elif isinstance(value, str):
82
- self.value = None
+ value = None
83
elif isinstance(value, (list, tuple)):
84
- self.value = [float(v) for v in value]
85
- else:
86
- self.value = value
+ value = [float(v) for v in value]
+
+ self.value: Optional[Sequence[float]] = value
87
self.inplace = inplace
88
89
self._log_ratio = torch.log(torch.tensor(self.ratio))
0 commit comments