MlflowCallback

MlflowCallback

Определение

MlflowCallback — это класс-колбэк из библиотеки HuggingFace Transformers, используемый для интеграции с MLflow. При передаче в Trainer он автоматически логирует метрики обучения (loss, accuracy и др.), параметры конфигурации, а также артефакты (например, веса модели) в ходе fine-tuning. Это позволяет централизованно отслеживать эксперименты, сравнивать результаты разных запусков и воспроизводить обучение.

Где встречается

Навигация