Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
def __init__(self, inplanes, planes, stride=1, downsample=None, noactivation=False):
super(RevBasicBlock, self).__init__()
if downsample is None and stride == 1:
gm = BasicBlockSub(inplanes // 2, planes // 2, stride, noactivation)
fm = BasicBlockSub(inplanes // 2, planes // 2, stride, noactivation)
self.revblock = ReversibleBlock(gm, fm)
else:
self.basicblock_sub = BasicBlockSub(inplanes, planes, stride, noactivation)
self.downsample = downsample
self.stride = stride
def __init__(self, inplanes, planes, stride=1, downsample=None, noactivation=False):
super(RevBasicBlock, self).__init__()
if downsample is None and stride == 1:
gm = BasicBlockSub(inplanes // 2, planes // 2, stride, noactivation)
fm = BasicBlockSub(inplanes // 2, planes // 2, stride, noactivation)
self.revblock = ReversibleBlock(gm, fm)
else:
self.basicblock_sub = BasicBlockSub(inplanes, planes, stride, noactivation)
self.downsample = downsample
self.stride = stride
def __init__(self, inplanes, planes, stride=1, downsample=None, noactivation=False):
super(RevBasicBlock, self).__init__()
if downsample is None and stride == 1:
gm = BasicBlockSub(inplanes // 2, planes // 2, stride, noactivation)
fm = BasicBlockSub(inplanes // 2, planes // 2, stride, noactivation)
self.revblock = ReversibleBlock(gm, fm)
else:
self.basicblock_sub = BasicBlockSub(inplanes, planes, stride, noactivation)
self.downsample = downsample
self.stride = stride
def __init__(self, inplanes, planes, stride=1, noactivation=False):
super(BasicBlockSub, self).__init__()
self.noactivation = noactivation
if not self.noactivation:
self.bn1 = batch_norm(inplanes)
self.conv1 = conv3x3(inplanes, planes, stride)
self.bn2 = batch_norm(planes)
self.conv2 = conv3x3(planes, planes)
self.relu = nn.ReLU(inplace=True)