Skip to content

Commit e2b982e

Browse files
Update monai/losses/perceptual.py
Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com> Signed-off-by: Virginia Fernandez <[email protected]>
1 parent 2156b84 commit e2b982e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

monai/losses/perceptual.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -131,7 +131,7 @@ def __init__(
131131
net=network_type, verbose=False, channel_wise=channel_wise, cache_dir=cache_dir
132132
)
133133
elif "radimagenet_" in network_type:
134-
self.perceptual_function = RadImageNetPerceptualSimilarity(net=network_type, verbose=False)
134+
self.perceptual_function = RadImageNetPerceptualSimilarity(net=network_type, verbose=False, cache_dir=cache_dir)
135135
elif network_type == "resnet50":
136136
self.perceptual_function = TorchvisionModelPerceptualSimilarity(
137137
net=network_type,

0 commit comments

Comments
 (0)