Browse Source

Update tools.py

main
BBing 2 years ago
parent
commit
61f25a1553
1 changed files with 0 additions and 1 deletions
  1. +0
    -1
      utils/tools.py

+ 0
- 1
utils/tools.py View File

@@ -3,7 +3,6 @@ import mindspore.numpy as mnp
import mindspore import mindspore
from mindspore import Tensor, Parameter from mindspore import Tensor, Parameter



def adjust_learning_rate(optimizer, epoch, args): def adjust_learning_rate(optimizer, epoch, args):
if args.lradj == 'type1': if args.lradj == 'type1':
lr_adjust = {epoch: args.learning_rate * (0.5 ** ((epoch-1) // 1))} lr_adjust = {epoch: args.learning_rate * (0.5 ** ((epoch-1) // 1))}


Loading…
Cancel
Save