相关代码在tensorflow源码中:
site-packages\tensorflow\contrib\slim\python\slim\learning.py中
def train()--------->负责程序运行整体流程
...
logging.info('Starting Session.')
...
logging.info('Starting Queues.')
...
try:
while not sv.should_stop():
total_loss, should_stop = train_step_fn(
sess, train_op, global_step, train_step_kwargs)
if should_stop:
logging.info('Stopping Training.')
sv.request_stop()
break
...
if logdir and sv.is_chief:
logging.info('Finished training! Saving model to disk.')
...
def train_step()-------->控制台输出训练信息
if 'should_log' in train_step_kwargs:
if sess.run(train_step_kwargs['should_log']):
logging.info('global step %d: loss = %.4f (%.3f sec/step)',
np_global_step, total_loss, time_elapsed)