Keras + Tensorflow model convert to coreml exits NameError: global name ... is not defined

Question

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

NameError: global name `batch_size' is not defined

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

非常感谢，

J.

Answer 1

我运行在使用参数构造神经网络时变成了类似的信息。这应该有效：

from keras import models

batch_size = 50    

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

tensorflow