如何在十位数内恢复模型

Question

首先我用tf.contrib.gan我像吼叫一样，能训练模特。

tf.contrib.gan.gan_train(
        train_ops,
        hooks=(
                [tf.train.StopAtStepHook(num_steps=FLAGS.max_number_of_steps),
                 tf.train.LoggingTensorHook([status_message], every_n_iter=10)] +
                sync_hooks),
        logdir=FLAGS.train_log_dir,
        master=FLAGS.master,
        is_chief=FLAGS.task == 0,
        config=conf
    )

然后我想评估模型。当试图以以下方式恢复检查点时

^{pr2}$

我得到了这个例外：

^{pr3}$

我使用TF 1.7rc1

Answer 1

实际上，生成的图形中有一个问题。这些是我为了解决这个问题而采取的步骤。在

第一步： 使用以下代码打印checkpoint文件中的所有变量

from tensorflow.python.tools.inspect_checkpoint import print_tensors_in_checkpoint_file
print_tensors_in_checkpoint_file(file_name, '')

Step2:然后我注意到每个键都包含第一个作用域（“Discriminator”）的重复项，但当我试图加载模型时，它并不包含该部分。因此，我不得不用下面的方法去掉这个额外的部分

^{pr2}$

第3步：然后我可以成功地加载模型，如下所示。在

^{3}$

希望这对可能遇到同样问题的人有所帮助。在