[输入参数类型不一致]torchvision.models.mobilenet_v3_small

torchvision.models.mobilenet_v3_small

torchvision.models.mobilenet_v3_small(*, weights: Optional[MobileNet_V3_Small_Weights] = None, progress: bool = True, **kwargs: Any)

paddle.vision.models.mobilenet_v3_small

paddle.vision.models.mobilenet_v3_small(pretrained=False, scale=1.0, **kwargs)

两者功能一致但参数类型不一致，具体如下：

参数映射

torchvision PaddlePaddle 备注
weights pretrained 预训练权重，PyTorch 参数 weights 为 MobileNet_V3_Small_Weights 枚举类或 String 类型，Paddle 参数 pretrained 为 bool 类型，需要转写。
progress - 是否显示下载进度条，Paddle 无此参数，一般对网络训练结果影响不大，可直接删除。
- scale 通道数缩放比例，PyTorch 无此参数，Paddle 保持默认即可。
kwargs kwargs 附加的关键字参数。

转写示例

weights: 预训练权重

# PyTorch 写法
torchvision.models.mobilenet_v3_small(weights=torchvision.models.MobileNet_V3_Small_Weights.DEFAULT)

# Paddle 写法
paddle.vision.models.mobilenet_v3_small(pretrained=True)