Skip to content

DPOTrainer generate_during_eval support for MLFlow #3659

@dhruvmullick

Description

@dhruvmullick

Feature request

Like Comet and Wandb, also have MLFlow support for the generate_during_eval parameter

https://github.com/huggingface/trl/blob/main/trl/trainer/dpo_trainer.py#L1802-L1831

Motivation

Have parity with MLFlow, an open source framework for visualizing training runs

Your contribution

Will submit a PR

Metadata

Metadata

Assignees

No one assigned

    Labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions