tensorflow 保存的PB模型200+MB 怎么处理 5C

下面的是模型保存密码的部分

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()

1个回答

Csdn user default icon
上传中...
上传图片
插入图片
抄袭、复制答案,以达到刷声望分或其他目的的行为,在CSDN问答是严格禁止的,一经发现立刻封号。是时候展现真正的技术了!
立即提问