昇腾社区首页
中文
注册

tft_train

接口功能

经过MindIO TTP装饰器包装的训练框架train方法,捕获训练状态异常以及上报处理,目前仅对ModelLink框架适用,其他框架供参考。

接口格式

mindio_ttp.adaptor.tft_train(train_args, test_data_iterator_list)

接口参数

参数

是否必选

说明

取值要求

train_args

必选

ModelLink框架build_train_args方法返回第一个参数。

list,由build_train_args方法决定。

test_data_iterator_list

必选

ModelLink框架build_train_args方法返回第二个参数。

list,由build_train_args方法决定。

返回值

ModelLink的train方法。