mirror of
https://github.com/hiyouga/LLaMA-Factory.git
synced 2025-08-02 03:32:50 +08:00
tiny fix
Former-commit-id: 55027282cdaa59a470ac89bfb3860504ba9075ff
This commit is contained in:
parent
413a206652
commit
236f97b35c
@ -31,6 +31,12 @@ Dependency graph:
|
||||
transformers>=4.41.2,<=4.44.3
|
||||
packing:
|
||||
transformers>=4.41.2,<=4.44.3
|
||||
|
||||
Disable version checking: DISABLE_VERSION_CHECK=1
|
||||
Enable VRAM recording: RECORD_VRAM=1
|
||||
Force check imports: FORCE_CHECK_IMPORTS=1
|
||||
Force using torchrun: FORCE_TORCHRUN=1
|
||||
Use modelscope: USE_MODELSCOPE_HUB=1
|
||||
"""
|
||||
|
||||
from .extras.env import VERSION
|
||||
|
@ -318,7 +318,7 @@ class LogCallback(TrainerCallback):
|
||||
if self.webui_mode and all(key in logs for key in ["loss", "learning_rate", "epoch"]):
|
||||
logger.info(
|
||||
"{{'loss': {:.4f}, 'learning_rate': {:2.4e}, 'epoch': {:.2f}, 'throughput': {}}}".format(
|
||||
logs["loss"], logs["learning_rate"], logs["epoch"], logs.get("throughput")
|
||||
logs["loss"], logs["learning_rate"], logs["epoch"], logs.get("throughput", "N/A")
|
||||
)
|
||||
)
|
||||
|
||||
|
Loading…
x
Reference in New Issue
Block a user