|
@@ -331,14 +331,14 @@ def main(_):
|
|
|
train_input = PTBInput(config=config, data=train_data, name="TrainInput")
|
|
train_input = PTBInput(config=config, data=train_data, name="TrainInput")
|
|
|
with tf.variable_scope("Model", reuse=None, initializer=initializer):
|
|
with tf.variable_scope("Model", reuse=None, initializer=initializer):
|
|
|
m = PTBModel(is_training=True, config=config, input_=train_input)
|
|
m = PTBModel(is_training=True, config=config, input_=train_input)
|
|
|
- tf.contrib.deprecated.scalar_summary("Training Loss", m.cost)
|
|
|
|
|
- tf.contrib.deprecated.scalar_summary("Learning Rate", m.lr)
|
|
|
|
|
|
|
+ tf.scalar_summary("Training Loss", m.cost)
|
|
|
|
|
+ tf.scalar_summary("Learning Rate", m.lr)
|
|
|
|
|
|
|
|
with tf.name_scope("Valid"):
|
|
with tf.name_scope("Valid"):
|
|
|
valid_input = PTBInput(config=config, data=valid_data, name="ValidInput")
|
|
valid_input = PTBInput(config=config, data=valid_data, name="ValidInput")
|
|
|
with tf.variable_scope("Model", reuse=True, initializer=initializer):
|
|
with tf.variable_scope("Model", reuse=True, initializer=initializer):
|
|
|
mvalid = PTBModel(is_training=False, config=config, input_=valid_input)
|
|
mvalid = PTBModel(is_training=False, config=config, input_=valid_input)
|
|
|
- tf.contrib.deprecated.scalar_summary("Validation Loss", mvalid.cost)
|
|
|
|
|
|
|
+ tf.scalar_summary("Validation Loss", mvalid.cost)
|
|
|
|
|
|
|
|
with tf.name_scope("Test"):
|
|
with tf.name_scope("Test"):
|
|
|
test_input = PTBInput(config=eval_config, data=test_data, name="TestInput")
|
|
test_input = PTBInput(config=eval_config, data=test_data, name="TestInput")
|