初始PyTorch(六+):ResNet18的网络结构

使用ResNet18网络结构,为了更好适配Cifar-10数据集【h*w=32*32】,因此不是彻底按照ResNet18的参数写的。python 下图是ResNet18的内部结构图。网络 先写内部结构:有两层的weight layer。ide class ResBlk(nn.Module): def __init__(self, ch_in, ch_out, stride=1):
相关文章
相关标签/搜索