Skip to content

GitLab

Unverified Commit f554f2d6 authored by joerg-de's avatar joerg-de Committed by GitHub
Browse files

added fill to __repr__ of RandomRotation (#2340)

parent b6f55ed8
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
torchvision/transforms/transforms.py
View file @ f554f2d6
...@@ -1081,6 +1081,8 @@ class RandomRotation(object): ...@@ -1081,6 +1081,8 @@ class RandomRotation(object):
format_string += ', expand={0}'.format(self.expand) format_string += ', expand={0}'.format(self.expand)
if self.center is not None: if self.center is not None:
format_string += ', center={0}'.format(self.center) format_string += ', center={0}'.format(self.center)
if self.fill is not None:
format_string += ', fill={0}'.format(self.fill)
format_string += ')' format_string += ')'
return format_string return format_string
......
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