slim.learning.train()

代码语言：javascript复制

slim.learning.train(train_op, logdir, train_step_fn=train_step,
                    train_step_kwargs=_USE_DEFAULT,
                    log_every_n_steps=1, graph=None, master='',
                    is_chief=True, global_step=None,
                    number_of_steps=None, init_op=_USE_DEFAULT,
                    init_feed_dict=None, local_init_op=_USE_DEFAULT,
                    init_fn=None, ready_op=_USE_DEFAULT,
                    summary_op=_USE_DEFAULT,
                    save_summaried_secs=600,
                    summary_writer=_USE_DEFAULT,
                    startup_delay_steps=0, saver=None,
                    save_interval_secs=600, sync_optimizer=None,
                    session_config=None, session_wrapper=None,
                    trace_every_n_steps=None,
                    ignore_live_threads=False)

参数众多，其中重要的有：

• train_op，指定优化算法
• logdir，指定训练数据保存文件夹
• save_summaries_secs，指定每隔多少秒更新一次日志文件（对应 tensorboard 刷新一次的时间）
• save_interval_secs，指定每隔多少秒保存一次模型

tensorboard 模型 日志 算法 优化

0 人点赞