Skip to content

Latest commit

 

History

History
27 lines (20 loc) · 705 Bytes

densenet.rst

File metadata and controls

27 lines (20 loc) · 705 Bytes

DenseNet

.. currentmodule:: torchvision.models

The DenseNet model is based on the Densely Connected Convolutional Networks paper.

Model builders

The following model builders can be used to instantiate a DenseNet model, with or without pre-trained weights. All the model builders internally rely on the torchvision.models.densenet.DenseNet base class. Please refer to the source code for more details about this class.

.. autosummary::
    :toctree: generated/
    :template: function.rst

    densenet121
    densenet161
    densenet169
    densenet201