Alexander Borzunov 3 лет назад
Родитель
Сommit
5db0a43b25
1 измененных файлов с 1 добавлено и 1 удалено
  1. 1 1
      examples/albert/arguments.py

+ 1 - 1
examples/albert/arguments.py

@@ -148,4 +148,4 @@ class AlbertTrainingArguments(TrainingArguments):
 
     save_total_limit: int = 2
     save_steps: int = 500
-    max_steps: int = 10 ** 30
+    max_steps: int = 10 ** 30  # meant as "peer should compute gradients forever"