(in_planes, out_planes)
| 106 | |
| 107 | |
| 108 | def UpBlockComp(in_planes, out_planes): |
| 109 | block = nn.Sequential( |
| 110 | nn.Upsample(scale_factor=2, mode='nearest'), |
| 111 | conv2d(in_planes, out_planes*2, 3, 1, 1, bias=False), |
| 112 | #convTranspose2d(in_planes, out_planes*2, 4, 2, 1, bias=False), |
| 113 | NoiseInjection(), |
| 114 | batchNorm2d(out_planes*2), GLU(), |
| 115 | conv2d(out_planes, out_planes*2, 3, 1, 1, bias=False), |
| 116 | NoiseInjection(), |
| 117 | batchNorm2d(out_planes*2), GLU() |
| 118 | ) |
| 119 | return block |
| 120 | |
| 121 | |
| 122 | class Generator(nn.Module): |
no test coverage detected