ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fixing logging for each step

Browse source
This commit is contained in:
Manan17 2026-03-10 20:32:10 +00:00
commit 1bede34409
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
studio/backend/core/training/worker.py
View file

@ -609,7 +609,7 @@ def _run_embedding_training(event_queue: Any, stop_queue: Any, config: dict) ->
"learning_rate": lr_value,
"fp16": not is_bfloat16_supported(),
"bf16": is_bfloat16_supported(),
"logging_steps": max(1, log_frequency) if log_frequency else 1,
"logging_steps": 1,
"report_to": ["wandb"] if config.get("enable_wandb") else "none",
"lr_scheduler_type": config.get("lr_scheduler_type", "linear"),
"batch_sampler": BatchSamplers.NO_DUPLICATES,