TensorFlow量化训练

  1. 云栖社区>
  2. 博客>
  3. 正文

TensorFlow量化训练

烁凡 2018-12-20 10:08:35 浏览4821
展开阅读全文
前段时间研究了tflite和量化相关的操作, 经测试量化尤其在具有专门DSP加速的硬件上(比如MTK8183)有着很好的加速效果,大约3X的提升;
tensorflow提供了tflite转化工具toco,使用命令大致如下:
bazel-bin/tensorflow/contrib/lite/toco/toco --input_file=mobilenet_v1_1.0_128_frozen.pb \
  --input_format=TENSORFLOW_GRAPHDEF \
  --output_format=TFLITE \
  --output_file=/tmp/mobilenet_v1_1.0_128.tflite \
  --inference_type=FLOAT \
  --input_data_types=FLOAT

网友评论

登录后评论
0/500
评论
烁凡
+ 关注