Commit a7ba27b1 authored by yzy5630's avatar yzy5630
Browse files

add parser for adam

parent d6522e28
...@@ -159,6 +159,10 @@ def main(): ...@@ -159,6 +159,10 @@ def main():
default=0, default=0,
type=int, type=int,
help="Linear warmup over warmup_steps.") 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", parser.add_argument("--learning_rate",
default=3e-5, default=3e-5,
type=float, type=float,
......
...@@ -434,6 +434,10 @@ def main(): ...@@ -434,6 +434,10 @@ def main():
default=3e-5, default=3e-5,
type=float, type=float,
help="The initial learning rate for Adam.") 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", parser.add_argument("--num_train_epochs",
default=3.0, default=3.0,
type=float, type=float,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment