-
Notifications
You must be signed in to change notification settings - Fork 1.4k
Description
Hi,
I'd like to build a customized SegResNet, where the blocks have different kernel sizes (according to a paper that I found, which handles feature extraction in a similar case as mine with these special kernel sizes).
However, if I construct a network using ResBlocks, I cannot change the kernel sizes, although there is an arg for that:
resblock = monai.networks.blocks.ResBlock( spatial_dims=3, in_channels=8, norm=('GROUP', {'num_groups': 4}), kernel_size=5, act=('RELU', {'inplace': True}) )
Output:
ResBlock(
(norm1): GroupNorm(4, 8, eps=1e-05, affine=True)
(norm2): GroupNorm(4, 8, eps=1e-05, affine=True)
(act): ReLU(inplace=True)
(conv1): Convolution(
(conv): Conv3d(8, 8, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1), bias=False)
)
(conv2): Convolution(
(conv): Conv3d(8, 8, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1), bias=False)
)
)
Note, that although I passed a kernel_size = 5 arg, they are still size 3. Any hints? Thanks!
Originally posted by @r00tdotexe in Project-MONAI/tutorials#1524