tf第五讲:模型保存与加载(tf.train.Saver()&tf.saved_model)及fine_tune(梯度冻结)

文章目录

  • 1. tf.train.Saver()
    • 1.1 说明
    • 1.2 模型保存
    • 1.3 模型的加载
    • 1.4 fine_tune(冻结梯度)
  • 2. tf.save_model
    • 2.1 模型保存
    • 2.2 构造signatureDef protobuf
    • 2.3 简单案例
    • 2.4 保存模型生成的文件目录
    • 2.5 DNN回归案例
    • 2.6 模型加载
      • 2.6.1 加载步骤
      • 2.6.2 基于简单案例的模型加载
      • 2.6.3 基于DNN回归案例的模型加载

1. tf.train.Saver()

1.1 说明

  • 该种方式保存模型,会生成

你可能感兴趣的:(tf第五讲:模型保存与加载(tf.train.Saver()&tf.saved_model)及fine_tune(梯度冻结))