Skip to content

Add pretrain option for DenseNet and SENet #1804

@yiheng-wang-nv

Description

@yiheng-wang-nv

So far, Class DenseNet and Senet do not have an option to load pretrained weights for specific arch. Users need to call a function, such as densenet121 and se_resnet50 to load pretrained weights. It may not be very convenient for some cases that only Class type are supported to call.
Let me submit a PR to improve it.

Metadata

Metadata

Labels

enhancementNew feature or request

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions