在使用 CIFAR-10 训练网络的 TensorFlow 教程中，它们在哪里以及如何在运行训练和评估之间保存权重/参数？我看不到任何保存到我的项目目录的文件。
以下是教程和代码的链接： https://www.tensorflow.org/versions/r0.11/tutorials/deep_cnn/index.html https://github.com/tensorflow/tensorflow/tree/master/tensorflow/models/image/cifar10
【问题讨论】：
标签： tensorflow
默认将日志和检查点保存到/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 的项目，我必须更改它，因为主硬盘没有足够的空间来容纳 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 中，恢复权重发生在这一行：
【讨论】：