(
in_channels,
zq_ch,
add_conv,
gather=False,
)
| 412 | |
| 413 | |
| 414 | def Normalize3D( |
| 415 | in_channels, |
| 416 | zq_ch, |
| 417 | add_conv, |
| 418 | gather=False, |
| 419 | ): |
| 420 | return SpatialNorm3D( |
| 421 | in_channels, |
| 422 | zq_ch, |
| 423 | gather=gather, |
| 424 | # norm_layer=nn.GroupNorm, |
| 425 | freeze_norm_layer=False, |
| 426 | add_conv=add_conv, |
| 427 | num_groups=32, |
| 428 | eps=1e-6, |
| 429 | affine=True, |
| 430 | ) |
| 431 | |
| 432 | |
| 433 | class Upsample3D(nn.Module): |