File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
mmgen/models/architectures/ddpm Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -112,9 +112,9 @@ class DenoisingUnet(nn.Module):
112112 downsample_conv (bool, optional): Whether use conv operation in
113113 downsample block. Defaults to ``True``.
114114 upsample_cfg (dict, optional): Config for upsample blocks.
115- Defaults to ``dict(type='DenoisingDownsample')``.
116- downsample_cfg (dict, optional): Config for downsample blocks.
117115 Defaults to ``dict(type='DenoisingUpsample')``.
116+ downsample_cfg (dict, optional): Config for downsample blocks.
117+ Defaults to ``dict(type='DenoisingDownsample')``.
118118 attention_res (int | list[int], optional): Resolution of feature maps
119119 to apply attention operation. Defaults to ``[16, 8]``.
120120 pretrained (str | dict, optional): Path for the pretrained model or
You can’t perform that action at this time.
0 commit comments