Unverified Commit 8265469b authored by ProGamerGov's avatar ProGamerGov Committed by GitHub
Browse files

Correct make_grid's pad_value variable type hint (#5182)

parent 3e79d149
......@@ -19,7 +19,7 @@ def make_grid(
normalize: bool = False,
value_range: Optional[Tuple[int, int]] = None,
scale_each: bool = False,
pad_value: int = 0,
pad_value: float = 0.0,
**kwargs,
) -> torch.Tensor:
"""
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment