代码之家  ›  专栏  ›  技术社区  ›  jbm

Keras+Tensorflow模型转换为coreml出口名称错误:全局名称…未定义

  •  0
  • jbm  · 技术社区  · 7 年前

    NameError: global name `batch_size' is not defined
    

    非常感谢,

    J

    1 回复  |  直到 7 年前
        1
  •  1
  •   sn0wbl1nd    6 年前

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

    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