首页 > 解决方案 > Keras 模型未保存 - 未创建文件

问题描述

我真的找不到任何方法来生成文件。没有权限问题,没有错误......但没有文件输出......

我的例子如下。

def build_model(self):
    model = tf.keras.Sequential([
        tf.keras.layers.Dense(128, activation='relu'),
        tf.keras.layers.Dense(128, activation='relu'),
        tf.keras.layers.Dense(5, activation='sigmoid')
    ])

    model.build([1, 256, 256, 3])

    return model

然后我打电话:

 self.build_model().save('~/Documents/model')

但是,那里没有任何输出,我不知道为什么这个简单的示例不输出任何文件......这一切都封装在我在 pytest 中运行的测试中

标签: pythontensorflowkeras

解决方案


推荐阅读