Question

我已经从keras网站调整了VAE示例来训练我的数据，一切运行正常。但是我无法转换为coreml。错误是：

NameError: global name `batch_size' is not defined

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

非常感谢，

学家

Answer 1

在使用参数构造神经网络时，我遇到了类似的消息。这应该起作用：

from keras import models

batch_size = 50    

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