张量如何在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"
它也适用于其它例子。
答
让我们假设你正在运行cifar10_train,节省发生在这条线:
而默认位置定义在这一行(它是“/ tmp/cifar10_train”):
在cifar10_eval,恢复权重上这一行情况:
请考虑添加评论,如果你认为这个职位可以改善。 –