Constructor.
1003 ):
1004 super(GBlock, self).__init__()
1005
1006
1007 self.in_channels, self.out_channels = in_channels, out_channels
1008
1009 self.hidden_channels = self.in_channels // channel_ratio
1010
1011 self.which_conv, self.which_bn = which_conv, which_bn
1012
1013 self.activation = activation
1014
1015
1016 self.conv1 = self.which_conv(
1017 self.in_channels, self.hidden_channels, kernel_size=1, padding=0
1018 )
1019
1020 self.conv2 = self.which_conv(self.hidden_channels, self.hidden_channels)
1021
1022 self.conv3 = self.which_conv(self.hidden_channels, self.hidden_channels)
1023
1024 self.conv4 = self.which_conv(
1025 self.hidden_channels, self.out_channels, kernel_size=1, padding=0
1026 )
1027
1028
1029 self.bn1 = self.which_bn(self.in_channels)
1030
1031 self.bn2 = self.which_bn(self.hidden_channels)
1032
1033 self.bn3 = self.which_bn(self.hidden_channels)
1034
1035 self.bn4 = self.which_bn(self.hidden_channels)
1036
1037 self.upsample = upsample
1038