diff --git a/utils/tools.py b/utils/tools.py index 3930442..61508f1 100644 --- a/utils/tools.py +++ b/utils/tools.py @@ -3,6 +3,7 @@ import mindspore.numpy as mnp import mindspore from mindspore import Tensor, Parameter + def adjust_learning_rate(optimizer, epoch, args): if args.lradj == 'type1': lr_adjust = {epoch: args.learning_rate * (0.5 ** ((epoch-1) // 1))}