pytorch densenet121
时间: 2023-11-12 07:55:30 浏览: 130
densenet121.py例子
DenseNet-121是一种在PyTorch中实现的深度学习模型,121表示它有121个层。这包括120个卷积层和1个全连接层。在构建DenseNet-121时,可以使用_densenet()函数,它会调用DenseNet()来构建模型。如果需要使用预训练模型,_densenet()函数还会负责加载预训练模型。在DenseBlock中,每个DenseLayer的输入特征图的通道数会依次增加growth_rate(对于DenseNet-121来说是32)。
阅读全文