Skip to content

Move pretrained_weights to convnext and swint configs. #154

@gitttt-1234

Description

@gitttt-1234

Convnext and SwinT backbones support loading pre-trained ImageNet weights from torchvision.models repository. To initialize the model with these weights, we use the model_config.pretrained_weights parameter. However, since this pretrained_weights parameter is only supported for convnext and swint backbones, it would be more structures to move it under model_config.convnext and model_config.swint sections.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions