如何像我们在 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)
如何在本机张量流中做同样的事情