HybridTrainPipelineSparseDist.progress
功能描述
进行流水训练。
函数原型
1 | def progress(dataloader_iter: Iterator[In]) -> Out: |
参数说明
参数名 |
类型 |
可选/必选 |
说明 |
|---|---|---|---|
dataloader_iter |
Iterator[In] |
必选 |
数据集迭代器。该迭代器返回用于查表和训练的Batch类，参考1。 |
返回值说明
- 成功：返回模型的输出。
- 失败：抛出异常。
使用示例
1 | pipeline.progress(batched_iterator) |
父主题： pipeline接口