Tensorflow元文件被删除

时间:2018-07-03 14:55:15

标签: tensorflow

我是tensorflow的新手。我希望在每k个全局步骤后保存一次模型,但是我想在开始时只保存一次模型的图元文件(其中包含图结构的详细信息)。我正在尝试以下代码,但是几次更新后我的图元文件将被删除。有人可以告诉我我可能做错了什么吗?

with tf.Session() as sess:
      saver.save(sess, MODELSAVEDIR2+'DNN_model_metadata')
      try:
          while True:
            _, step = sess.run([train_op, global_step])                                 
            if step % 1000 == 0:
               saver.save(sess,MODELSAVEDIR+'DNN_model'+MODELFILESTR, 
               global_step = step,write_meta_graph=False)
      except tf.errors.OutOfRangeError:
          pass

0 个答案:

没有答案