add parser for adam
This commit is contained in:
1 parent
d6522e2873
commit
a7ba27b1b4
2 files changed
+8
No files matched your search
@@ -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,
|
||||
|
||||
Reference in new issue
Block a user