Skip to content

GitLab

Commit 4a450b25 authored by tholor's avatar tholor
Browse files

removing unused argument eval_batch_size from LM finetuning #256

parent 58f0a274
Show whitespace changes
Inline Side-by-side
Showing with 0 additions and 4 deletions
examples/run_lm_finetuning.py
View file @ 4a450b25
...@@ -432,10 +432,6 @@ def main(): ...@@ -432,10 +432,6 @@ def main():
default=32, default=32,
type=int, type=int,
help="Total batch size for training.") help="Total batch size for training.")
parser.add_argument("--eval_batch_size",
default=8,
type=int,
help="Total batch size for eval.")
parser.add_argument("--learning_rate", parser.add_argument("--learning_rate",
default=3e-5, default=3e-5,
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