tensorflow - Keras + Tensorflow 模型转换为 coreml 退出 NameError: global name ... is not defined

Question

我已经改编了 keras 网站上的 VAE 示例来训练我的数据，并且一切运行良好。但我无法转换为 coreml。错误是：

NameError: global name `batch_size' is not defined

由于在 python 源代码中明确定义了 batch_size，我猜它与转换工具如何捕获变量名有关。有谁知道我可以如何修复它（或者它是否确实可以修复）？

非常感谢，

J。

score 1 · Accepted Answer

在使用参数构建神经网络时，我遇到了类似的信息。这应该有效：

from keras import models

batch_size = 50    

model = models.load_model(filename, custom_objects={'batch_size': batch_size})

1 回答 1