mindflow.common

接口名

概述

支持平台

mindflow.common.get_multi_step_lr

epoch的数量到达其中一个milestone时,学习率按 gamma 进行衰减,生成学习率数组。

Ascend GPU CPU

mindflow.common.get_poly_lr

生成指数衰减的学习率。

Ascend GPU

mindflow.common.get_warmup_cosine_annealing_lr

基于余弦函数生成衰减学习率数组。

Ascend GPU CPU