PaddleX在模型训练时,存在以下两种情况需要进行联网下载
1.训练模型时,用户没有配置自定义的预训练模型权重
pretrain_weights,此时PaddleX会自动联网下载在标准数据集上的预训练模型; 2.模型裁剪训练时,用户没有配置自定义的参数敏感度信息文件sensitivities_file,并将sensitivities_file配置成了'DEFAULT'字符串,此时PaddleX会自动联网下载模型在标准数据集上计算得到的参数敏感度信息文件。
在训练模型时,不管是正常训练还是裁剪训练,用户可以提前准备好预训练权重或参数敏感度信息文档,只需自定义
pretrain_weights或sensitivities_file, 将其设为本地的路径即可。
以下模型均为分类模型权重(UNet除外),用户在训练模型时,需要根据分类模型的种类或backbone的种类,选择对应的模型权重进行下载(目标检测在使用ResNet50作为Backbone时,使用下面表格中的ResNet50_cos作为预训练模型)
| 模型(点击下载) | 数据集 |
|---|---|
| ResNet18 | ImageNet |
| ResNet34 | ImageNet |
| ResNet50 | ImageNet |
| ResNet101 | ImageNet |
| ResNet50_vd | ImageNet |
| ResNet101_vd | ImageNet |
| MobileNetV1 | ImageNet |
| MobileNetV2_x1.0 | ImageNet |
| MobileNetV2_x0.5 | ImageNet |
| MobileNetV2_x2.0 | ImageNet |
| MobileNetV2_x0.25 | ImageNet |
| MobileNetV2_x1.5 | ImageNet |
| MobileNetV3_small | ImageNet |
| MobileNetV3_large | ImageNet |
| DarkNet53 | ImageNet |
| DenseNet121 | ImageNet |
| DenseNet161 | ImageNet |
| DenseNet201 | ImageNet |
| ResNet50_cos | ImageNet |
| Xception41 | ImageNet |
| Xception65 | ImageNet |
| ShuffleNetV2 | ImageNet |
| UNet | MSCOCO |