Skip to content

Commit 7ce29a1

Browse files
committed
Fix network_type docstring formatting in PerceptualLoss
1 parent 15fd428 commit 7ce29a1

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

monai/losses/perceptual.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -49,9 +49,11 @@ class PerceptualLoss(nn.Module):
4949
5050
Args:
5151
spatial_dims: number of spatial dimensions.
52-
network_type: {``"alex"``, ``"vgg"``, ``"squeeze"``, ``"radimagenet_resnet50"``,
53-
``"medicalnet_resnet10_23datasets"``, ``"medicalnet_resnet50_23datasets"``, ``"resnet50"``}
54-
Specifies the network architecture to use. Defaults to ``"alex"``.
52+
network_type: str or PercetualNetworkType Specifies the network architecture to use. Supported values are defined in
53+
``PercetualNetworkType`` (e.g., ``"alex"``, ``"vgg"``, ``"squeeze"``,
54+
``"radimagenet_resnet50"``, ``"medicalnet_resnet10_23datasets"``,
55+
``"medicalnet_resnet50_23datasets"``, ``"resnet50"``).
56+
Defaults to ``PercetualNetworkType.alex``.
5557
is_fake_3d: if True use 2.5D approach for a 3D perceptual loss.
5658
fake_3d_ratio: ratio of how many slices per axis are used in the 2.5D approach.
5759
cache_dir: path to cache directory to save the pretrained network weights.

0 commit comments

Comments
 (0)