This function upsamples input tensor by a factor of 2
(x)
| 32 | |
| 33 | |
| 34 | def upsample(x): |
| 35 | """ |
| 36 | This function upsamples input tensor by a factor of 2 |
| 37 | """ |
| 38 | return F.interpolate(x, scale_factor=2, mode='nearest') |
| 39 | |
| 40 | |
| 41 | def conv2d(in_planes, out_planes, kernel_size=3, stride=1, dilation=1, nonlin = 'LRU', padding_mode = 'reflect', norm = False): |