Skip to content
This repository was archived by the owner on Feb 7, 2025. It is now read-only.

Commit 49e5545

Browse files
authored
Fix format (#349)
Signed-off-by: Walter Hugo Lopez Pinaya <ianonimato@hotmail.com>
1 parent 6c5037a commit 49e5545

3 files changed

Lines changed: 5 additions & 10 deletions

File tree

generative/metrics/ms_ssim.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -121,7 +121,7 @@ def _compute_metric(self, y_pred: torch.Tensor, y: torch.Tensor) -> torch.Tensor
121121
avg_pool = getattr(F, f"avg_pool{self.spatial_dims}d")
122122

123123
multiscale_list: list[torch.Tensor] = []
124-
for i in range(len(weights)):
124+
for _ in range(len(weights)):
125125
ssim, cs = compute_ssim_and_cs(
126126
y_pred=y_pred,
127127
y=y,

tutorials/generative/image_to_image_translation/tutorial_segmentation_with_ddpm.ipynb

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -182,7 +182,7 @@
182182
"from monai.apps import DecathlonDataset\n",
183183
"from monai.config import print_config\n",
184184
"from monai.data import DataLoader\n",
185-
"from monai.utils import first, set_determinism\n",
185+
"from monai.utils import set_determinism\n",
186186
"from torch.cuda.amp import GradScaler, autocast\n",
187187
"from tqdm import tqdm\n",
188188
"\n",
@@ -693,9 +693,7 @@
693693
"metadata": {},
694694
"outputs": [],
695695
"source": [
696-
"scheduler = DDPMScheduler(\n",
697-
" num_train_timesteps=1000,\n",
698-
")\n",
696+
"scheduler = DDPMScheduler(num_train_timesteps=1000)\n",
699697
"optimizer = torch.optim.Adam(params=model.parameters(), lr=2.5e-5)\n",
700698
"inferer = DiffusionInferer(scheduler)"
701699
]
@@ -1462,7 +1460,7 @@
14621460
"plt.imshow(var[0, ...].cpu(), vmin=0, vmax=1, cmap=\"jet\") # We plot the variance map\n",
14631461
"plt.tight_layout()\n",
14641462
"plt.axis(\"off\")\n",
1465-
"plt.show()\n"
1463+
"plt.show()"
14661464
]
14671465
}
14681466
],

tutorials/generative/image_to_image_translation/tutorial_segmentation_with_ddpm.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -184,9 +184,7 @@
184184
)
185185
model.to(device)
186186

187-
scheduler = DDPMScheduler(
188-
num_train_timesteps=1000,
189-
)
187+
scheduler = DDPMScheduler(num_train_timesteps=1000)
190188
optimizer = torch.optim.Adam(params=model.parameters(), lr=2.5e-5)
191189
inferer = DiffusionInferer(scheduler)
192190

@@ -396,4 +394,3 @@ def dice_coeff(im1, im2, empty_score=1.0):
396394
plt.tight_layout()
397395
plt.axis("off")
398396
plt.show()
399-

0 commit comments

Comments
 (0)