Skip to content

Commit 3e92d74

Browse files
committed
Fix PerceptualLoss enum references and docstrings
1 parent 84e7d70 commit 3e92d74

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
@@ -49,7 +49,7 @@ class PerceptualLoss(nn.Module):
4949
5050
Args:
5151
spatial_dims: number of spatial dimensions.
52-
network_type : str
52+
network_type: str | PercetualNetworkType = PercetualNetworkType.alex,
5353
One of:
5454
- "alex"
5555
- "vgg"

0 commit comments

Comments
 (0)