add parser for adam
This commit is contained in:
@@ -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():
|
|||||||
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,
|
||||||
|
|||||||
Reference in New Issue
Block a user