我已经从keras网站调整了VAE示例来训练我的数据,一切运行正常 . 但是我无法转换为coreml . 错误是:
NameError: global name `batch_size' is not defined
由于batch_size明确地在python源中定义,我猜它与转换工具如何捕获变量名称有关 . 有谁知道我如何解决它(或者是否确实可以修复)?
非常感谢,
J.
当使用参数构建神经网络时,我遇到了类似的消息 . 这应该工作:
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
1 回答
当使用参数构建神经网络时,我遇到了类似的消息 . 这应该工作:
另见文档:https://keras.io/getting-started/faq/#how-can-i-save-a-keras-model