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

torchvision.models.shufflenet_v2_x0_5

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

paddle.vision.models.shufflenet_v2_x0_5

paddle.vision.models.shufflenet_v2_x0_5(pretrained=False, **kwargs)

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

参数映射

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

转写示例

weights: 预训练权重

# PyTorch 写法
torchvision.models.shufflenet_v2_x0_5(weights=torchvision.models.ShuffleNet_V2_X0_5_Weights.DEFAULT)

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