Browse Source

fix mass lr import.

tags/v1.1.0
linqingke 5 years ago
parent
commit
f26003e723
1 changed files with 3 additions and 1 deletions
  1. +3
    -1
      model_zoo/official/nlp/mass/src/utils/lr_scheduler.py

+ 3
- 1
model_zoo/official/nlp/mass/src/utils/lr_scheduler.py View File

@@ -17,7 +17,9 @@ from math import ceil
import numpy as np import numpy as np


import mindspore.nn.learning_rate_schedule as lr_schedules import mindspore.nn.learning_rate_schedule as lr_schedules

from mindspore import Tensor
from mindspore.ops import operations as P
import mindspore.common.dtype as mstype


def square_root_schedule(lr, update_num, decay_start_step, def square_root_schedule(lr, update_num, decay_start_step,
warmup_steps=2000, warmup_steps=2000,


Loading…
Cancel
Save