[Pytorch] AlexNet神經網路結構 ... self.dense = nn.Sequential(. nn.Linear(9216, 4096),. nn.ReLU(),. nn.Dropout(0.5), ... out = self.dense(res). return out ...
確定! 回上一頁