Skip to content

Fix network_type docstring formatting in PerceptualLoss #8670

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
aman0311x wants to merge 13 commits into
base: dev
Choose a base branch
from
Open

Fix network_type docstring formatting in PerceptualLoss #8670

Changes from 2 commits
Commits
Show all changes
13 commits
Select commit Hold shift + click to select a range
7ce29a1
Fix network_type docstring formatting in PerceptualLoss
aman0311x Dec 22, 2025
e2d0a02
DCO Remediation Commit for Mohammad Amanour Rahman <amanourrahman@gma…
aman0311x Dec 22, 2025
96ea131
DCO Remediation Commit for Mohammad Amanour Rahman <amanourrahman@gma…
aman0311x Dec 22, 2025
cc0c951
Fix PerceptualLoss enum references and docstrings
aman0311x Dec 22, 2025
eeef172
DCO Remediation Commit for Mohammad Amanour Rahman <amanourrahman@gma…
aman0311x Dec 22, 2025
84e7d70
DCO Remediation Commit for Mohammad Amanour Rahman <amanourrahman@gma…
aman0311x Dec 22, 2025
3e92d74
Fix PerceptualLoss enum references and docstrings
aman0311x Dec 22, 2025
ea64afa
DCO Remediation Commit for Mohammad Amanour Rahman <amanourrahman@gma…
aman0311x Dec 22, 2025
0c667a9
DCO Remediation Commit for Mohammad Amanour Rahman <amanourrahman@gma…
aman0311x Dec 22, 2025
ccea893
DCO Remediation Commit for Mohammad Amanour Rahman <amanourrahman@gma…
aman0311x Dec 22, 2025
8be3faa
Fix network_type docstring formatting in PerceptualLoss
aman0311x Dec 22, 2025
f563b89
WIP: temp commit before cherry-pick
aman0311x Dec 22, 2025
a234edf
Mohammad Amanour Rahman <[email protected]>
aman0311x Dec 22, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
14 changes: 10 additions & 4 deletions monai/losses/perceptual.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,7 @@
torchvision, _ = optional_import("torchvision")


class PercetualNetworkType(StrEnum):
class PerceptualNetworkType(StrEnum):
alex = "alex"
vgg = "vgg"
squeeze = "squeeze"
Expand All @@ -49,9 +49,15 @@ class PerceptualLoss(nn.Module):

Args:
spatial_dims: number of spatial dimensions.
network_type: {``"alex"``, ``"vgg"``, ``"squeeze"``, ``"radimagenet_resnet50"``,
``"medicalnet_resnet10_23datasets"``, ``"medicalnet_resnet50_23datasets"``, ``"resnet50"``}
Specifies the network architecture to use. Defaults to ``"alex"``.
network_type : str
One of:
- "alex"
- "vgg"
- "squeeze"
- "radimagenet_resnet50"
- "medicalnet_resnet10_23datasets"
- "medicalnet_resnet50_23datasets"
- "resnet50"
is_fake_3d: if True use 2.5D approach for a 3D perceptual loss.
fake_3d_ratio: ratio of how many slices per axis are used in the 2.5D approach.
cache_dir: path to cache directory to save the pretrained network weights.
Expand Down
Loading