指数decay位于另一个具有嵌入权重的图中

2024-10-02 02:44:38 发布

您现在位置:Python中文网/ 问答频道 /正文

建立一个指数衰减模型,并开始训练模型。但是,我们收到了错误消息

Tensor("ExponentialDecay:0", shape=(), dtype=float32) 
  must be from the same graph as Tensor("dnn/.../embedding_weights/...", ...)

看起来根本原因是,在训练过程中,tf试图创建另一个图形,而不是创建张量“指数decay”时使用的图形。但这应该如何发生呢

learning_rate = tf.train.exponential_decay(learning_rate=0.1, global_step=tf.train.get_or_create_global_step(), decay_steps=100, decay_rate=0.96)

optimizer = tf.train.AdagradOptimizer(learning_rate=learning_rate)

estimator = tf.estimator.DNNClassifier(
    feature_columns=...,
    hidden_units=...,
    n_classes=...,
    model_dir=...,
    optimizer=optimizer)

tf.estimator.train_and_evaluate(
    estimator=estimator,
    train_spec=...,
    eval_spec=...)

Tags: 模型图形ratetfsteptrain指数global

热门问题