From 5516e9193c035181528d6e33e20dbf09d95240b4 Mon Sep 17 00:00:00 2001 From: Philip Meier Date: Wed, 15 Feb 2023 13:38:07 +0100 Subject: [PATCH] add error for max_size with size sequence in resize --- torchvision/prototype/transforms/functional/_geometry.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/torchvision/prototype/transforms/functional/_geometry.py b/torchvision/prototype/transforms/functional/_geometry.py index 814697f03a3..543a3bbc725 100644 --- a/torchvision/prototype/transforms/functional/_geometry.py +++ b/torchvision/prototype/transforms/functional/_geometry.py @@ -148,6 +148,11 @@ def _compute_resized_output_size( ) -> List[int]: if isinstance(size, int): size = [size] + elif max_size is not None and len(size) != 1: + raise ValueError( + "max_size should only be passed if size specifies the length of the smaller edge, " + "i.e. size should be an int or a sequence of length 1 in torchscript mode." + ) return __compute_resized_output_size(spatial_size, size=size, max_size=max_size)