下面的是模型保存密码的部分
def save_model(sess, epoch):
builder = tf.saved_model.builder.SavedModelBuilder("model-v1.0_%d" % epoch)
builder.add_meta_graph_and_variables(sess, ['v1.0'])
builder.save()
下面的是模型保存密码的部分
def save_model(sess, epoch):
builder = tf.saved_model.builder.SavedModelBuilder("model-v1.0_%d" % epoch)
builder.add_meta_graph_and_variables(sess, ['v1.0'])
builder.save()