用tensorflow创建文件目录以及删除目录下所有内容

最新推荐文章于 2020-05-07 22:06:07 发布

Hanx_22

最新推荐文章于 2020-05-07 22:06:07 发布

阅读量1.7k

点赞数

CC 4.0 BY-SA版权

分类专栏： tensorflow

本文链接：https://blog.youkuaiyun.com/Hanx_22/article/details/77248495

tensorflow 专栏收录该内容

7 篇文章

订阅专栏

本文介绍了一个使用TensorFlow进行模型训练检查点管理的简单示例。通过定义路径和配置是否从检查点恢复来控制模型训练过程。如果指定路径不存在，则会创建相应目录；若路径已存在且不恢复，则先清空目录再创建。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#coding=utf-8
import tensorflow as tf

tf.app.flags.DEFINE_string('checkpoint_path', 'E:/E/test/src/checkpoint', '')
tf.app.flags.DEFINE_boolean('restore', False, 'whether to resotre from checkpoint')

FLAGS = tf.app.flags.FLAGS

print(FLAGS.checkpoint_path)
if not tf.gfile.Exists(FLAGS.checkpoint_path):
        tf.gfile.MkDir(FLAGS.checkpoint_path) #创建目录
else:
    if not FLAGS.restore:
        tf.gfile.DeleteRecursively(FLAGS.checkpoint_path) #删除该目录下的全部内容
        tf.gfile.MkDir(FLAGS.checkpoint_path)