摘要: 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 阅读全文
posted @ 2021-09-03 17:59 哈哈哈喽喽喽 阅读(40) 评论(0) 推荐(0) 编辑
摘要: class BasicBlockGroup(nn.Module): expansion = 1 def __init__(self, in_planes, planes, stride=1, groups=2, bn=False): super(BasicBlockGroup, self).__in 阅读全文
posted @ 2021-09-03 17:58 哈哈哈喽喽喽 阅读(8) 评论(0) 推荐(0) 编辑
摘要: class Bottleneck(nn.Module): expansion = 4 def __init__(self, in_planes, planes, stride=1): super(Bottleneck, self).__init__() self.conv1 = nn.Conv2d( 阅读全文
posted @ 2021-09-03 17:58 哈哈哈喽喽喽 阅读(26) 评论(0) 推荐(0) 编辑
摘要: class BasicBlock(nn.Module): expansion = 1 def __init__(self, in_planes, planes, stride=1, bn=False): super(BasicBlock, self).__init__() self.conv1 = 阅读全文
posted @ 2021-09-03 17:57 哈哈哈喽喽喽 阅读(7) 评论(0) 推荐(0) 编辑