Browse Source

decouple max steps from scheduler steps

justheuristic 3 years ago
parent
commit
04408e2ac7
1 changed files with 6 additions and 2 deletions
  1. 6 2
      examples/albert/arguments.py

+ 6 - 2
examples/albert/arguments.py

@@ -138,10 +138,14 @@ class AlbertTrainingArguments(TrainingArguments):
     fp16: bool = True
     fp16_opt_level: str = "O2"
     do_train: bool = True
+    do_eval: bool = False
 
+    logging_dir: str = "logs"
+    output_dir: str = "outputs"
     logging_steps: int = 100
+    logging_first_step: bool = True
+    overwrite_output_dir: bool = True
+
     save_total_limit: int = 2
     save_steps: int = 500
     max_steps: int = 10 ** 30
-
-    output_dir: str = "outputs"