Unverified Commit 6d774c6f authored by vfdev's avatar vfdev Committed by GitHub
Browse files

Fixed repr for ElasticTransform (#6758)


Co-authored-by: default avatarVasilis Vryniotis <datumbox@users.noreply.github.com>
parent b16dec19
...@@ -2133,9 +2133,9 @@ class ElasticTransform(torch.nn.Module): ...@@ -2133,9 +2133,9 @@ class ElasticTransform(torch.nn.Module):
return F.elastic_transform(tensor, displacement, self.interpolation, self.fill) return F.elastic_transform(tensor, displacement, self.interpolation, self.fill)
def __repr__(self): def __repr__(self):
format_string = self.__class__.__name__ + "(alpha=" format_string = self.__class__.__name__
format_string += str(self.alpha) + ")" format_string += f"(alpha={self.alpha}"
format_string += ", (sigma=" + str(self.sigma) + ")" format_string += f", sigma={self.sigma}"
format_string += ", interpolation={self.interpolation}" format_string += f", interpolation={self.interpolation}"
format_string += ", fill={self.fill})" format_string += f", fill={self.fill})"
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