added fill to __repr__ of RandomRotation (#2340)

added fill to repr of RandomRotation (#2340)
f554f2d6 · joerg-de · GitHub · b6f55ed8 · f554f2d6
Unverified Commit f554f2d6 authored Jun 23, 2020 by joerg-de Committed by GitHub Jun 23, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

torchvision/transforms/transforms.py torchvision/transforms/transforms.py +2 -0

No files found.
--- a/torchvision/transforms/transforms.py
+++ b/torchvision/transforms/transforms.py
@@ -1081,6 +1081,8 @@ class RandomRotation(object):
        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