machine-learning - 从经过训练的 RNN 模型生成单词：“变量已经存在，不允许。你的意思是在 VarScope 中设置 reuse=True 吗？”

Question

所以我在 jupytor notebook 中实现了一个 RNN 词生成器模型。当我尝试使用经过训练的模型生成一些单词时：

    with open(os.path.join(cfgs['save_dir'], 'config.pkl'), 'rb') as f:
       saved_args = cPickle.load(f)

    with open(os.path.join(cfgs['save_dir'], 'words_vocab.pkl'), 'rb') as f:
       words, vocab = cPickle.load(f)

    with tf.Session() as sess:
       model = Model(saved_args, True)
       tf.global_variables_initializer().run()
       saver = tf.train.Saver(tf.global_variables())
       ckpt = tf.train.get_checkpoint_state(cfgs['save_dir'])
       if ckpt and ckpt.model_checkpoint_path:
           saver.restore(sess, ckpt.model_checkpoint_path)
           print(model.sample(sess, words, vocab, cfgs['n'], cfgs['prime'], cfgs['sample'], cfgs['pick'], cfgs['width']))

它第一次工作，但如果我再次运行代码，则会出现错误：

    ValueError: Variable rnnlm/softmax_w already exists, disallowed. Did you mean to set reuse=True in VarScope? 

现在我必须关闭 ipynb 文件，然后运行代码以获取新示例。如何更改代码以避免这种情况？

Answer 1

您可以多次调用 model.sample 函数而不会出现问题，但其他一切（创建会话、构建模型、加载检查点）只应运行一次。如果您重构代码，那么您将不会再看到该错误消息。