Lighteval 文件
EvaluationTracker
加入 Hugging Face 社群
並獲得增強的文件體驗
開始使用
EvaluationTracker
class lighteval.logging.evaluation_tracker.EvaluationTracker
< 原始碼 >( output_dir: str save_details: bool = True push_to_hub: bool = False push_to_tensorboard: bool = False hub_results_org: str | None = '' tensorboard_metric_prefix: str = 'eval' public: bool = False nanotron_run_info: GeneralArgs = None wandb: bool = False )
引數
- output_dir (
str
) — 您希望儲存結果的本地資料夾路徑。 - save_details (
bool
, 預設為 True) — 如果為 True,詳細資訊將儲存到output_dir
。 - push_to_hub (
bool
, 預設為 False) — 如果為 True,詳細資訊將被推送到 Hub。對於模型model_name
,結果將被推送到{hub_results_org}/details__{sanitized model_name}
(如果public
為 True,則為公共資料集),或{hub_results_org}/details__{sanitized model_name}_private
(私有資料集)。 - push_to_tensorboard (
bool
, 預設為 False) — 如果為 True,將在 Hub 上建立並推送結果到一個 TensorBoard 資料夾。 - hub_results_org (
str
, 可選) — 推送結果的目標組織。有關資料集組織的更多詳細資訊,請參閱EvaluationTracker.save
。 - tensorboard_metric_prefix (
str
, 預設為 “eval”) — TensorBoard 日誌中指標的字首。 - public (
bool
, 預設為 False) — 如果為 True,結果和詳細資訊將被推送到公共組織。 - nanotron_run_info (
~nanotron.config.GeneralArgs
, 可選) — 引用有關 Nanotron 模型執行的資訊。
跟蹤整個評估過程和相關資訊。
EvaluationTracker 包含用於實驗詳細資訊(DetailsLogger)、指標(MetricsLogger)、任務版本(VersionsLogger)以及特定任務(TaskConfigLogger)和整個評估執行(GeneralConfigLogger)的通用配置的特定記錄器。它編譯來自這些記錄器的資料並將其寫入檔案,如果需要,可以釋出到 Hugging Face Hub。
屬性:
- details_logger (DetailsLogger) — 用於實驗詳細資訊的記錄器。
- metrics_logger (MetricsLogger) — 用於實驗指標的記錄器。
- versions_logger (VersionsLogger) — 用於任務版本的記錄器。
- general_config_logger (GeneralConfigLogger) — 用於通用配置的記錄器。
- task_config_logger (TaskConfigLogger) — 用於任務配置的記錄器。
將實驗詳細資訊(每個步驟的所有模型預測)推送到 Hub。
完全更新當前評估模型的詳細資訊儲存庫元資料卡
將實驗資訊和結果儲存到檔案,如果需要,還儲存到 Hub。