Skip to content

Suggesting updates on the documentation of DenseNet #20144

Closed
@ParsifalXu

Description

@ParsifalXu

Hi,

It seems that there is a potential incomplete description of the document of DenseNet(Similar issues also exist in Xception and ResNet). As mentioned in the document:

classes: optional number of classes to classify images into, only to be specified if include_top is True, and if no weights argument is specified.

However, corresponding part in the code:

if weights == "imagenet" and include_top and classes != 1000:
    raise ValueError(
        'If using `weights` as `"imagenet"` with `include_top`'
        " as true, `classes` should be 1000"
    )

Although this will not cause a bug, because the default value is 1000, I think this default value should be added like similar documents(MobileNet) to reduce unnecessary understanding.

Metadata

Metadata

Labels

type:docsNeed to modify the documentation

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions