diff --git a/PaddleNLP/pretrain_language_models/XLNet/run_classifier.py b/PaddleNLP/pretrain_language_models/XLNet/run_classifier.py index 1c58e39f30aa7e4fa64857058f178ffb78b12392..44184b763d364c77e2c066274433599fe2fcf41e 100755 --- a/PaddleNLP/pretrain_language_models/XLNet/run_classifier.py +++ b/PaddleNLP/pretrain_language_models/XLNet/run_classifier.py @@ -69,7 +69,7 @@ init_g.add_arg("init_std", str, 0.02, "Initialization std when init is norm init_g.add_arg("init_range", str, 0.1, "Initialization std when init is uniform.") train_g = ArgumentGroup(parser, "training", "training options.") -train_g.add_arg("epoch", int, 3, "Number of epoches for fine-tuning.") +train_g.add_arg("epoch", int, 1000, "Number of epoches for fine-tuning.") train_g.add_arg("learning_rate", float, 5e-5, "Learning rate used to train with warmup.") train_g.add_arg("lr_scheduler", str, "linear_warmup_decay", "scheduler of learning rate.", choices=['linear_warmup_decay', 'noam_decay'])