From 94682f36ed06aa99512b7fc611b7d1cc8f230c3c Mon Sep 17 00:00:00 2001 From: Liufang Sang Date: Mon, 21 Oct 2019 14:58:29 +0800 Subject: [PATCH] fix change ResNet50 to ResNet34 in doc test=release/1.6 (#3683) --- .../classification/quantization/README.md | 21 +++++++++++++------ 1 file changed, 15 insertions(+), 6 deletions(-) diff --git a/PaddleSlim/classification/quantization/README.md b/PaddleSlim/classification/quantization/README.md index 7119fef3..4c475220 100644 --- a/PaddleSlim/classification/quantization/README.md +++ b/PaddleSlim/classification/quantization/README.md @@ -40,7 +40,7 @@ cost = fluid.layers.cross_entropy(input=out, label=label) - use_gpu: 是否使用gpu。如果选择使用GPU，请确保当前环境和Paddle版本支持GPU。默认为True。 - batch_size: 在量化之后，对模型进行fine-tune训练时用的batch size。 -- model: 要压缩的目标模型，该示例支持'MobileNet', 'MobileNetV2'和'ResNet50'。 +- model: 要压缩的目标模型，该示例支持'MobileNet', 'MobileNetV2'和'ResNet34'。 - pretrained_model: 预训练模型的路径，可以从[这里](https://github.com/PaddlePaddle/models/tree/develop/PaddleCV/image_classification#%E5%B7%B2%E5%8F%91%E5%B8%83%E6%A8%A1%E5%9E%8B%E5%8F%8A%E5%85%B6%E6%80%A7%E8%83%BD)下载。 - config_file: 压缩策略的配置文件。 @@ -196,14 +196,23 @@ fluid.optimizer.Momentum(momentum=0.9, >训练超参： -### ResNet50 +优化器 +``` +fluid.optimizer.Momentum(momentum=0.9, + learning_rate=fluid.layers.piecewise_decay( + boundaries=[5000 * 12], + values=[0.0001, 0.00001]), + regularization=fluid.regularizer.L2Decay(1e-4)) +``` +8卡，batch size 1024，epoch 30, 挑选好的结果 +### ResNet34 | weight量化方式 | activation量化方式| top1_acc/top5_acc |Paddle Fluid inference time(ms)| Paddle Lite inference time(ms)|模型下载| |---|---|---|---|---|---| -|baseline|- |76.50%/93.00%|- |-|[下载模型](http://paddle-imagenet-models-name.bj.bcebos.com/ResNet50_pretrained.tar)| -|abs_max|abs_max|76.71%/93.10% |- |-|[下载模型](https://paddle-slim-models.bj.bcebos.com/quantization%2Fresnet50_w_abs_a_abs_7670_9310.tar.gz)| -|abs_max|moving_average_abs_max|76.65%/93.12% |- |-|[下载模型](https://paddle-slim-models.bj.bcebos.com/quantization%2Fresnet50_w_abs_a_move_7665_9312.tar.gz) | -|channel_wise_abs_max|abs_max|76.56%/93.05% |- |-| [下载模型](https://paddle-slim-models.bj.bcebos.com/quantization%2Fresnet50_w_chan_a_abs_7656_9304.tar.gz)| +|baseline|- |74.57%/92.14%|- |-|-| +|abs_max|abs_max||- |-|-| +|abs_max|moving_average_abs_max||- |-|-| +|channel_wise_abs_max|abs_max||- |-| -| >训练超参： -- GitLab