diff --git a/dgmr/common.py b/dgmr/common.py index 70b20fc..2cf5e70 100644 --- a/dgmr/common.py +++ b/dgmr/common.py @@ -271,7 +271,6 @@ def __init__( ) def forward(self, x) -> torch.Tensor: - if self.input_channels < self.output_channels: sc = self.conv_1x1(x) sc = torch.cat([x, sc], dim=1) diff --git a/tests/test_losses.py b/tests/test_losses.py index b5f27d4..d456261 100644 --- a/tests/test_losses.py +++ b/tests/test_losses.py @@ -4,7 +4,6 @@ def test_ssim_loss(): - x = torch.rand((2, 3, 32, 32)) y = torch.rand((2, 3, 32, 32)) @@ -17,7 +16,6 @@ def test_ssim_loss(): def test_ms_ssim_loss(): - x = torch.rand((2, 3, 256, 256)) y = torch.rand((2, 3, 256, 256)) @@ -30,7 +28,6 @@ def test_ms_ssim_loss(): def test_ssim_loss_dynamic(): - x = torch.rand((2, 3, 256, 256)) y = torch.rand((2, 3, 256, 256)) curr_image = torch.rand((2, 3, 256, 256)) @@ -44,7 +41,6 @@ def test_ssim_loss_dynamic(): def test_tv_loss(): - x = torch.ones((2, 3, 256, 256)) x[0, 0, 0, 0] = 2.5