Skip to content

Commit cbf5dc5

Browse files
author
Virginia Fernandez
committed
Fix default
1 parent c9bb8f7 commit cbf5dc5

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

monai/networks/nets/diffusion_model_unet.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1904,7 +1904,7 @@ def __init__(
19041904
spatial_dims: int,
19051905
in_channels: int,
19061906
out_channels: int,
1907-
input_shape: Sequence[int],
1907+
input_shape: Sequence[int] = (64, 64),
19081908
num_res_blocks: Sequence[int] | int = (2, 2, 2, 2),
19091909
channels: Sequence[int] = (32, 64, 64, 64),
19101910
attention_levels: Sequence[bool] = (False, False, True, True),

0 commit comments

Comments
 (0)