再定义一个ResNet网络
我们本次准备构建ResNet-18层结构
class ResNet(nn.Module):
def __init__(self):
super(ResNet...3, stride=1, padding=1),
nn.BatchNorm2d(64)
)
# 紧跟着要进行四次这样的单元
# 构建辅助函数...，使[b, 64, h, w] => [b, 128, h, w]
self.blk1 = ResBlk(64, 128)
# 构建辅助函数，使[b, 128, h, w...] = > [b, 256, h, w]
self.blk2 = ResBlk(128, 256)
# 构建辅助函数，使[b, 256, h, w] = > [b, 512...由于我们要进行10分类问题，要将添加代码
self.outlayer = nn.Linear(1024, 10)
和
x = self.outlayer(x)
return x
为确定具体维度大小，我们先构建假数据
云服务器
ICP备案
云直播
实时音视频
即时通信 IM
