mindflow.common

API Name

Description

Supported Platforms

mindflow.common.get_multi_step_lr

Generate decay learning rate array of each parameter group by gamma once the number of epoch reaches one of the milestones.

Ascend GPU CPU

mindflow.common.get_poly_lr

Generate polynomial decay learning rate array.

Ascend GPU

mindflow.common.get_warmup_cosine_annealing_lr

Calculates learning rate base on cosine decay function.

Ascend GPU CPU