diff --git a/README.md b/README.md index 528dffdac055d7e2447c300be7d37080d5cedf57..af7acb25cba844cc79a358744d50198e60c8de9d 100644 --- a/README.md +++ b/README.md @@ -57,6 +57,10 @@ softmax的计算公示如下图所示: 飞桨是由百度研发的一款源于产业实践的开源深度学习平台,致力于让深度学习技术的创新与应用更简单。PLSC基于飞桨平台研发,实现与飞桨平台的无缝链接,可以更好地服务产业实践。 +- 支持大规模分类 + + 单机8张V100 GPU配置下,支持的分类类别数增大了2.52倍; + - 包含多种预训练模型 除了PLSC库源码,我们还发布了基于ResNet50模型、ResNet101模型、ResNet152模型的大规模分类模型在多种数据集上的预训练模型,方便用户基于这些预训练模型进行下游任务的fine-tuning。 @@ -73,7 +77,7 @@ softmax的计算公示如下图所示: | 模型 | 描述 | | :--------------- | :------------- | -| [resnet50_distarcface_ms1m_v2](http://icm.baidu-int.com/user-center/account) | 该模型使用ResNet50网络训练,数据集为MS1M_v2,训练阶段使用的loss_type为'dist_arcface',预训练模型在lfw验证集上的验证精度为0.99817。 | +| [resnet50_distarcface_ms1m_arcface](https://plsc.bj.bcebos.com/pretrained_model/resnet50_distarcface_ms1mv2.tar.gz) | 该模型使用ResNet50网络训练,数据集为MS1M-ArcFace,训练阶段使用的loss_type为'dist_arcface',预训练模型在lfw验证集上的验证精度为0.99817。 | ### 训练性能 @@ -109,3 +113,10 @@ softmax的计算公示如下图所示: * [分布式参数转换](docs/distributed_params.md) * [Base64格式图像预处理](docs/base64_preprocessor.md) + +* 2019.12.23 + **`0.1.0`** + *PaddlePaddle大规模分类库(PLSC)发布,内建ResNet50、ResNet101和ResNet152三种模型,并支持自定义模型; + * 单机8张V100 GPU配置下,ResNet50模型一百万类别训练速度2,122.56 images/s, 并支持多机分布式训练; + * 发布模型在线预测库; + * 发布基于ResNet50网络和MS1M-ArcFace数据集的预训练模型。 diff --git a/plsc/models/resnet.py b/plsc/models/resnet.py index 1faffa27dc8d93461639f3dcb538987656e28375..df856513d7639db3b7bac0d4fed32c44d6b1e5aa 100644 --- a/plsc/models/resnet.py +++ b/plsc/models/resnet.py @@ -41,13 +41,11 @@ class ResNet(BaseModel): if layers == 50: depth = [3, 4, 14, 3] - num_filters = [64, 128, 256, 512] elif layers == 101: depth = [3, 4, 23, 3] - num_filters = [256, 512, 1024, 2048] elif layers == 152: depth = [3, 8, 36, 3] - num_filters = [256, 512, 1024, 2048] + num_filters = [64, 128, 256, 512] conv = self.conv_bn_layer( input=input, num_filters=64, filter_size=3, stride=1,