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

Question

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

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

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

Answer 1

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

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

不管怎样，解决方案都不是很好。