CANN/TensorFlow TellMeStepOrLossHook构造函数

TellMeStepOrLossHook构造函数【免费下载链接】tensorflowAscend TensorFlow Adapter项目地址: https://gitcode.com/cann/tensorflow功能说明TellMeStepOrLossHook类的构造函数TellMeStepOrLossHook用于告知底层软件“当前执行的步数和总的步数”或者“当前执行的loss和最终的目标loss”。函数原型class TellMeStepOrLossHook(session_run_hook.SessionRunHook): def __init__(self, stepNone, total_stepNone, lossNone, final_lossNone)参数说明参数名输入/输出描述step输入表示当前步数的Tensor的名称。total_step输入训练脚本总的训练步数。loss输入表示当前loss的Tensor的名称。final_loss输入训练脚本最终的目标loss。返回值返回TellMeStepOrLossHook类对象。约束说明Iterations_per_loop1的场景下会按照每增加Iterations_per_loop数量的步数就会告知底层软件当前执行的步数或者loss无法做到每增加1步就告知底层软件一次可能对底层软件某些依赖此hook函数结果的功能产生影响。调用示例from npu_bridge.npu_init import * est NPUEstimator( model_fnmodel_fn, configconfig, paramsparams) hooks [] max_steps 10000 # step分割的方式本示例当前step的tensor名称是global_step:0总step数是10000请根据实际step的tensor名称和总step数进行配置 my_hook TellMeStepOrLossHook(stepglobal_step:0, total_stepmax_steps) # loss分割的方式本示例当前loss的tensor名称是loss:0目标loss是7.1请根据实际loss的tensor名称和目标loss值进行配置 # my_hook TellMeStepOrLossHook(lossloss:0, final_loss7.1) hooks.append(my_hook) # 开启训练 est.train( input_fnimagenet_train.input_fn, max_stepsmax_steps hookshooks)【免费下载链接】tensorflowAscend TensorFlow Adapter项目地址: https://gitcode.com/cann/tensorflow创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考