摘要:
class BasicBlockGroup(nn.Module): expansion = 1 def __init__(self, in_planes, planes, stride=1, groups=2, bn=False): super(BasicBlockGroup, self).__in 阅读全文
摘要:
class Bottleneck(nn.Module): expansion = 4 def __init__(self, in_planes, planes, stride=1): super(Bottleneck, self).__init__() self.conv1 = nn.Conv2d( 阅读全文
摘要:
class BasicBlock(nn.Module): expansion = 1 def __init__(self, in_planes, planes, stride=1, bn=False): super(BasicBlock, self).__init__() self.conv1 = 阅读全文