Fedir-Ilina commited on
Commit
cc8390e
·
verified ·
1 Parent(s): c9c2b77

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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 = 2 # 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 = 2
562
  SAVE_STRATEGY = "steps"
563
- SAVE_STEPS = 2
564
- EVAL_STEPS = 2
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"