Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -554,14 +554,14 @@ if not globals().get("_LOGS_PRINTED", False):
|
|
| 554 |
BASE_BATCH_SIZE = 1 # Valor base para o batch size (reduzido para CPU)
|
| 555 |
INITIAL_ACCUMULATION_MIN_STEPS = 2
|
| 556 |
INITIAL_ACCUMULATION_MAX_STEPS = 4
|
| 557 |
-
BASE_EVAL_SIZE =
|
| 558 |
BASE_LEARNING_RATE = 5e-5 # 2e-5
|
| 559 |
LR_SCHEDULER_TYPE = "constant_with_warmup" # (HF: cosine + warmap) cosine
|
| 560 |
WARMUP_RATIO = 0.03 # ~3% dos 4012 steps ≈ 120
|
| 561 |
-
LOGGIN_STEPS =
|
| 562 |
SAVE_STRATEGY = "steps"
|
| 563 |
-
SAVE_STEPS =
|
| 564 |
-
EVAL_STEPS =
|
| 565 |
#MAX_GRAD_NORM = 1.0
|
| 566 |
WEIGHT_DECAY = 0.1
|
| 567 |
OPTIM = "adamw_torch"
|
|
|
|
| 554 |
BASE_BATCH_SIZE = 1 # Valor base para o batch size (reduzido para CPU)
|
| 555 |
INITIAL_ACCUMULATION_MIN_STEPS = 2
|
| 556 |
INITIAL_ACCUMULATION_MAX_STEPS = 4
|
| 557 |
+
BASE_EVAL_SIZE = 5 # Batch size para avaliação (reduzido para CPU)
|
| 558 |
BASE_LEARNING_RATE = 5e-5 # 2e-5
|
| 559 |
LR_SCHEDULER_TYPE = "constant_with_warmup" # (HF: cosine + warmap) cosine
|
| 560 |
WARMUP_RATIO = 0.03 # ~3% dos 4012 steps ≈ 120
|
| 561 |
+
LOGGIN_STEPS = 5
|
| 562 |
SAVE_STRATEGY = "steps"
|
| 563 |
+
SAVE_STEPS = 5
|
| 564 |
+
EVAL_STEPS = 5
|
| 565 |
#MAX_GRAD_NORM = 1.0
|
| 566 |
WEIGHT_DECAY = 0.1
|
| 567 |
OPTIM = "adamw_torch"
|