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

Question

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

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

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

Answer 1

很好，因为 keras API 在当前版本中不支持量化，所以您剩下 3 个选项：

1. 等待 Keras 具有所需的功能
2. 使用具有此功能的其他 API 重写模型
3. 查找不需要量化数据的其他 TPU

无论哪种方式，尽管解决方案都不是很好。