使用高级 keras 在 Tensorflow 中进行量化感知训练 api

Quantization-aware training in Tensorflow using the highlevel keras api

我使用 this colab 中描述的过程构建了我的第一个 covnet。现在我想 运行 Google 闪亮的新边缘 tpu 上的模型。

但是根据 here 描述的模型要求,我需要使用 quantization-aware training (post-training quantization is not supported). 才能将模型转换为我可以在 EdgeTPU 上使用的格式。

如何修改示例 colab 以执行此量化感知训练?

嗯,因为 keras API 在当前版本中不支持量化,你有 3 个选项:

  1. 等待 keras 具有所需的功能
  2. 使用具有此功能的不同 API 重写您的模型
  3. 寻找不需要您量化数据的不同 TPU

尽管解决方案不是很好。