摘要:
class ResNet5(nn.Module): # ResNet5(BasicBlock, [1, 1, 1, 1], 51) def init(self, block, num_blocks, num_classes=10, bn=False): super(ResNet5, self).in 阅读全文
摘要:
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 = 阅读全文