本文整理匯總了Python中train_utils.run_training方法的典型用法代碼示例。如果您正苦於以下問題：Python train_utils.run_training方法的具體用法？Python train_utils.run_training怎麽用？Python train_utils.run_training使用的例子？那麽, 這裏精選的方法代碼示例或許可以為您提供幫助。您也可以進一步了解該方法所在類train_utils的用法示例。
在下文中一共展示了train_utils.run_training方法的2個代碼示例，這些例子默認根據受歡迎程度排序。您可以為喜歡或者感覺有用的代碼點讚，您的評價將有助於係統推薦出更棒的Python代碼示例。
示例1: main
# 需要導入模塊: import train_utils [as 別名]
# 或者: from train_utils import run_training [as 別名]
def main(_):
"""Trains LSTM classification model."""
tf.logging.set_verbosity(tf.logging.INFO)
with tf.device(tf.train.replica_device_setter(FLAGS.ps_tasks)):
model = graphs.get_model()
train_op, loss, global_step = model.classifier_training()
train_utils.run_training(
train_op,
loss,
global_step,
variables_to_restore=model.pretrained_variables,
pretrained_model_dir=FLAGS.pretrained_model_dir) 示例2: main
# 需要導入模塊: import train_utils [as 別名]
# 或者: from train_utils import run_training [as 別名]
def main(_):
"""Trains Language Model."""
tf.logging.set_verbosity(tf.logging.INFO)
with tf.device(tf.train.replica_device_setter(FLAGS.ps_tasks)):
model = graphs.get_model()
train_op, loss, global_step = model.language_model_training()
train_utils.run_training(train_op, loss, global_step)