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

Question

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

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

如何修改示例colab来进行此量化感知训练？

Answer 1

好吧，因为keras API在当前版本中不支持量化，所以剩下3种选择：

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

无论哪种方法，解决方案都不是很好。