add parser for adam

This commit is contained in:
yzy5630 2019-07-18 08:52:51 +08:00
parent d6522e2873
commit a7ba27b1b4
2 changed files with 8 additions and 0 deletions

View File

@ -159,6 +159,10 @@ def main():
default=0,
type=int,
help="Linear warmup over warmup_steps.")
parser.add_argument("--adam_epsilon",
default=1e-8,
type=float,
help="Epsilon for Adam optimizer.")
parser.add_argument("--learning_rate",
default=3e-5,
type=float,

View File

@ -434,6 +434,10 @@ def main():
default=3e-5,
type=float,
help="The initial learning rate for Adam.")
parser.add_argument("--adam_epsilon",
default=1e-8,
type=float,
help="Epsilon for Adam optimizer.")
parser.add_argument("--num_train_epochs",
default=3.0,
type=float,