Shortcuts

框架.中间件.功能.训练器

训练器

训练师

ding.framework.middleware.functional.trainer.trainer(cfg: EasyDict, policy: Policy, log_freq: int = 100) Callable[source]
Overview:

执行单个训练过程的中间件。

Arguments:
  • cfg (EasyDict): 配置。

  • policy (Policy): 在逐步模式下要训练的策略。

  • log_freq (int): 显示日志的频率(迭代次数)。

多步训练器

ding.framework.middleware.functional.trainer.multistep_trainer(policy: Policy, log_freq: int = 100) Callable[source]
Overview:

为目标步骤数执行训练的中间件。

Arguments:
  • policy (Policy): 专门用于多步训练的策略。

  • log_freq (int): 显示日志的频率(迭代次数)。