File tree Expand file tree Collapse file tree 1 file changed +3
-4
lines changed
Expand file tree Collapse file tree 1 file changed +3
-4
lines changed Original file line number Diff line number Diff line change @@ -49,8 +49,7 @@ class PerceptualLoss(nn.Module):
4949
5050 Args:
5151 spatial_dims: number of spatial dimensions.
52- network_type: str | PercetualNetworkType = PercetualNetworkType.alex,
53- One of:
52+ network_type: type of network for perceptual loss. One of:
5453 - "alex"
5554 - "vgg"
5655 - "squeeze"
@@ -62,15 +61,15 @@ class PerceptualLoss(nn.Module):
6261 fake_3d_ratio: ratio of how many slices per axis are used in the 2.5D approach.
6362 cache_dir: path to cache directory to save the pretrained network weights.
6463 pretrained: whether to load pretrained weights. This argument only works when using networks from
65- LIPIS or Torchvision. Defaults to ``" True" ``.
64+ LIPIS or Torchvision. Defaults to ``True``.
6665 pretrained_path: if `pretrained` is `True`, users can specify a weights file to be loaded
6766 via using this argument. This argument only works when ``"network_type"`` is "resnet50".
6867 Defaults to `None`.
6968 pretrained_state_dict_key: if `pretrained_path` is not `None`, this argument is used to
7069 extract the expected state dict. This argument only works when ``"network_type"`` is "resnet50".
7170 Defaults to `None`.
7271 channel_wise: if True, the loss is returned per channel. Otherwise the loss is averaged over the channels.
73- Defaults to ``False``.
72+ Defaults to ``False``.
7473 """
7574
7675 def __init__ (
You can’t perform that action at this time.
0 commit comments