python - 使用高级 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

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