mindflow.core

API Name	Description	Supported Platforms
mindflow.core.AdaHessian	The Adahessian optimizer.	Ascend
mindflow.core.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.core.get_poly_lr	Generate polynomial decay learning rate array.	Ascend GPU
mindflow.core.get_warmup_cosine_annealing_lr	Calculates learning rate base on cosine decay function.	Ascend GPU CPU
mindflow.core.MTLWeightedLoss	Compute the MTL strategy weighted multi-task losses automatically.	Ascend GPU
mindflow.core.RelativeRMSELoss	Relative Root Mean Square Error (RRMSE) is the root mean squared error normalized by the root-mean-square value where each residual is scaled against the actual value.	Ascend GPU CPU
mindflow.core.WaveletTransformLoss	The multi-level wavelet transformation losses.	Ascend GPU
mindflow.core.get_loss_metric	Gets the loss function.	Ascend GPU
mindflow.core.batched_hessian	Calculate Hessian matrix of network model.	Ascend GPU CPU
mindflow.core.batched_jacobian	Calculate Jacobian matrix of network model.	Ascend GPU CPU