Skip to content

Commit

Permalink
replace "add" with "sub" in test
Browse files Browse the repository at this point in the history
  • Loading branch information
elad-c committed Jun 20, 2024
1 parent e2d6063 commit d35c5da
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -171,7 +171,7 @@ def forward(self, x):
_shape = x.shape[2:]
x = self.conv2d(input=x)
x2 = self.const_conv2d(input=self.conv_const)
x = torch.add(other=x2, input=x)
x = torch.sub(other=x2, input=x)
# input tensor in kwargs
x = nn.functional.interpolate(x, size=_shape)

Expand Down

0 comments on commit d35c5da

Please sign in to comment.