From 9d5f6aa1d1b39a820343e6280e7a4fda7b30f63f Mon Sep 17 00:00:00 2001 From: joerg-de <61417801+joerg-de@users.noreply.github.com> Date: Mon, 22 Jun 2020 22:35:04 +0200 Subject: [PATCH] added fill to __repr__ of RandomRotation --- torchvision/transforms/transforms.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/torchvision/transforms/transforms.py b/torchvision/transforms/transforms.py index d54aa5099f2..ba92bbccf6c 100644 --- a/torchvision/transforms/transforms.py +++ b/torchvision/transforms/transforms.py @@ -1081,6 +1081,8 @@ def __repr__(self): format_string += ', expand={0}'.format(self.expand) if self.center is not None: format_string += ', center={0}'.format(self.center) + if self.fill is not None: + format_string += ', fill={0}'.format(self.fill) format_string += ')' return format_string