初始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):
相關文章
相關標籤/搜索