Skip to content
This repository was archived by the owner on Feb 7, 2025. It is now read-only.

Commit eb01aee

Browse files
add test case
1 parent c6ed061 commit eb01aee

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

tests/test_perceptual_loss.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -14,9 +14,8 @@
1414
import unittest
1515

1616
import torch
17-
from parameterized import parameterized
18-
1917
from generative.losses import PerceptualLoss
18+
from parameterized import parameterized
2019

2120
TEST_CASES = [
2221
[{"spatial_dims": 2, "network_type": "squeeze"}, (2, 1, 64, 64), (2, 1, 64, 64)],
@@ -37,6 +36,11 @@
3736
(2, 1, 64, 64, 64),
3837
(2, 1, 64, 64, 64),
3938
],
39+
[
40+
{"spatial_dims": 3, "network_type": "resnet50", "is_fake_3d": True, "pretrained": True, "fake_3d_ratio": 0.2},
41+
(2, 1, 64, 64, 64),
42+
(2, 1, 64, 64, 64),
43+
],
4044
]
4145

4246

0 commit comments

Comments
 (0)