小编Sho*_*rma的帖子

Tensorflow 模型保存和加载

如何像我们在 keras 中所做的那样使用模型图保存张量流模型。我们可以保存整个模型(权重和图)并稍后导入,而不是在预测文件中再次定义整个图

在凯拉斯:

checkpoint = ModelCheckpoint('RightLane-{epoch:03d}.h5',monitor='val_loss', verbose=0,  save_best_only=False, mode='auto')
Run Code Online (Sandbox Code Playgroud)

将给出一个我们可以用于预测的 h5 文件

model = load_model("RightLane-030.h5")
Run Code Online (Sandbox Code Playgroud)

如何在本机张量流中做同样的事情

python deep-learning conv-neural-network keras tensorflow

5
推荐指数
1
解决办法
5879
查看次数