(*args, **kwargs)
| 17 | |
| 18 | |
| 19 | def WNConv2d(*args, **kwargs): |
| 20 | act = kwargs.pop("act", True) |
| 21 | conv = weight_norm(nn.Conv2d(*args, **kwargs)) |
| 22 | if not act: |
| 23 | return conv |
| 24 | return nn.Sequential(conv, nn.LeakyReLU(0.1)) |
| 25 | |
| 26 | |
| 27 | class MPD(nn.Module): |