diff --git a/demo/sequence-labeling/sequence_label.py b/demo/sequence-labeling/sequence_label.py index 9a028fe0fd28d6872f4a54c47f2230ee2002959f..f2871d136f802a41841c689d917e9a099fe1377d 100644 --- a/demo/sequence-labeling/sequence_label.py +++ b/demo/sequence-labeling/sequence_label.py @@ -73,7 +73,7 @@ if __name__ == '__main__': strategy = hub.AdamWeightDecayStrategy( weight_decay=args.weight_decay, learning_rate=args.learning_rate, - warmup_strategy="linear_warmup_decay", + lr_scheduler="linear_warmup_decay", ) # Setup runing config for PaddleHub Finetune API