昇腾社区首页
中文
注册
开发者
下载

progress

功能描述

执行训练流水线的一个步骤,包括前向传播、反向传播和参数更新。

函数原型

1
def progress(self, dataloader_iter: Iterator[In]) -> Out:

参数说明

参数名

类型

可选/必选

说明

dataloader_iter

Iterator[In]

必选

数据加载器迭代器

返回值说明

  • 成功:返回训练输出结果。
  • 失败:抛出StopIteration异常或RuntimeError。