当前位置: 首页 > 工具软件 > Slim Lang > 使用案例 >

slim.learning.train()

公孙森
2023-12-01

 参考 slim.learning.train() - 云+社区 - 腾讯云

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,指定每隔多少秒保存一次模型

 类似资料:

相关阅读

相关文章

相关问答