zoukankan      html  css  js  c++  java
  • tensorflow模型的保存与恢复

    模型保存后产生四个文件,分别是:

    |--models
    | |--checkpoint
    | |--.meta
    | |--.data
    | |--.index

    .meta保存的是图的结构

    checkpoint文件是个文本文件,里面记录了保存的最新的checkpoint文件以及其它checkpoint文件列表。

    .data和.index保存的是变量值。

    tensorflow常用的模型保存方法:

    best_str = ''
    if best_loss is None or valid_loss < best_loss:
      best_loss = valid_loss
      best_str = ', best!'
      early_stop_i = 0
      saver.save(sess, '%s/model_best.tf' % FLAGS.logdir)
    else:
      early_stop_i += 1

    代码比较简单,就是取每个轮次训练后损失最小的那一次保存模型,并进行后续加载

  • 相关阅读:
    基本排序
    mysql安装
    函数和方法
    COOKIE 与 SESSION
    django---入门
    django模板继承
    2017-11-14
    Django---模板层(template)
    Dajngo---model基础
    Django--基础篇
  • 原文地址:https://www.cnblogs.com/acm-icpcer/p/11686930.html
Copyright © 2011-2022 走看看