Skip to content

Commit f554f2d

Browse files
authored
added fill to __repr__ of RandomRotation (#2340)
1 parent b6f55ed commit f554f2d

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

torchvision/transforms/transforms.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1081,6 +1081,8 @@ def __repr__(self):
10811081
format_string += ', expand={0}'.format(self.expand)
10821082
if self.center is not None:
10831083
format_string += ', center={0}'.format(self.center)
1084+
if self.fill is not None:
1085+
format_string += ', fill={0}'.format(self.fill)
10841086
format_string += ')'
10851087
return format_string
10861088

0 commit comments

Comments
 (0)