张量如何在CIFAR10教程中保存张量流?

问题描述:

在TensorFlow教程中,在CIFAR-10上培训网络时,他们在哪里以及如何保存运行培训和评估之间的权重/参数?我看不到任何文件保存到我的项目目录。张量如何在CIFAR10教程中保存张量流?

这里是链接到的教程,代码: https://www.tensorflow.org/versions/r0.11/tutorials/deep_cnn/index.html https://github.com/tensorflow/tensorflow/tree/master/tensorflow/models/image/cifar10

它默认保存日志和检查站的/tmp/文件夹。 权重包含在检查点文件中。

正如您可以在eval和train文件中看到的那样,它将检查点目录作为参数。

cifar10_train.py:

tf.app.flags.DEFINE_string('train_dir', '/tmp/cifar10_train', 
          """Directory where to write event logs """ 
          """and checkpoint.""") 

cifar10_eval.py:

tf.app.flags.DEFINE_string('eval_dir', '/tmp/cifar10_eval', 
          """Directory where to write event logs.""") 
tf.app.flags.DEFINE_string('eval_data', 'test', 
          """Either 'test' or 'train_eval'.""") 
tf.app.flags.DEFINE_string('checkpoint_dir', '/tmp/cifar10_train', 
          """Directory where to read model checkpoints.""") 

你可以调用这些脚本与这些自定义值。对于我的使用Inception的项目,我必须改变它,因为主硬盘没有足够的空间来容纳最初创建的瓶颈。

由于/tmp/文件夹不是持久性的,因此您可能会丢失训练数据,因此明确设置这些值可能是一种很好的做法。

以下代码会将训练数据保存到自定义文件夹中。

python cifar10_train.py --train_dir="/home/username/train_folder" 

,然后评估:

python cifar10_eval.py --checkpoint_dir="/home/username/train_folder" 

它也适用于其它例子。

+0

请考虑添加评论,如果你认为这个职位可以改善。 –