added fill to __repr__ of RandomRotation (#2340)

pytorch · Jun 23, 2020 · f554f2d · f554f2d
1 parent b6f55ed
commit f554f2d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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