' ValueError: to_quantize 只能是 tf.keras 序列或功能模型

Question

我已经在我的模型训练脚本中尝试了 QAT 实现。我正在使用函数式 API 来创建模型。

我为实现 QAT API 所遵循的步骤，

1. 构建模型架构

2. 插入适当的 quantize_model 函数

3. 训练模型

让我为您提供代码片段以获得更多许可

std::function

版本详情

<块引用>

张量流==2.3.0

<块引用>

tensorflow-model-optimization==0.5.0

问题：我正在使用 QAT API 支持的函数式 API，但仍然出现值错误

<块引用>

quantize_model 'to_quantize 只能是 tf.keras 序列 或 ValueError: to_quantize 只能是 tf.keras Sequential 或 功能模型。

无法找出问题所在。如果有人帮助解决这个问题，那将会很有帮助。 提前致谢