我已经改编了 keras 网站上的 VAE 示例来训练我的数据，并且一切运行良好。但我无法转换为 coreml。错误是：
NameError: global name `batch_size' is not defined
由于在 python 源代码中明确定义了 batch_size，我猜它与转换工具如何捕获变量名有关。有谁知道我可以如何修复它（或者它是否确实可以修复）？
非常感谢，
J。
问问题
1157 次
1 回答
1
在使用参数构建神经网络时，我遇到了类似的信息。这应该有效：
from keras import models
batch_size = 50
model = models.load_model(filename, custom_objects={'batch_size': batch_size})
另请参阅文档：https ://keras.io/getting-started/faq/#how-can-i-save-a-keras-model
于 2018-09-26T20:51:25.220 回答