(in_channels)
| 59 | |
| 60 | |
| 61 | def Normalize(in_channels): |
| 62 | return torch.nn.GroupNorm(num_groups=32, num_channels=in_channels, eps=1e-6, affine=True) |
| 63 | |
| 64 | |
| 65 | # --------------------------------------------------------------------------------------------------- |