Skip to content
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 3 additions & 3 deletions tests/metrics/test_ssim_metric.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@

class TestSSIMMetric(unittest.TestCase):

def test2d_gaussian(self):
def test_2d_gaussian(self):
set_determinism(0)
preds = torch.abs(torch.randn(2, 3, 16, 16))
target = torch.abs(torch.randn(2, 3, 16, 16))
Expand All @@ -34,7 +34,7 @@ def test2d_gaussian(self):
expected_value = 0.045415
self.assertTrue(expected_value - result.item() < 0.000001)
Comment thread
ericspod marked this conversation as resolved.
Outdated
Comment thread
ericspod marked this conversation as resolved.
Outdated

def test2d_uniform(self):
def test_2d_uniform(self):
set_determinism(0)
preds = torch.abs(torch.randn(2, 3, 16, 16))
target = torch.abs(torch.randn(2, 3, 16, 16))
Expand All @@ -47,7 +47,7 @@ def test2d_uniform(self):
expected_value = 0.050103
self.assertTrue(expected_value - result.item() < 0.000001)
Comment thread
ericspod marked this conversation as resolved.
Outdated

def test3d_gaussian(self):
def test_3d_gaussian(self):
set_determinism(0)
preds = torch.abs(torch.randn(2, 3, 16, 16, 16))
target = torch.abs(torch.randn(2, 3, 16, 16, 16))
Expand Down
Loading