TRL

You are viewing main version, which requires installation from source. If you'd like regular pip install, checkout the latest stable version (v1.17.1).

Join the Hugging Face community

and get access to the augmented documentation experience

Collaborate on models, datasets and Spaces

Faster examples with accelerated inference

Switch between documentation themes

to get started

MergeModelCallback

فئة trl.experimental.merge_model_callback.MergeModelCallback

< source >

( merge_config: MergeConfig | None = Nonemerge_at_every_checkpoint: bool = Falsepush_to_hub: bool = False )

المعاملات

merge_config (experimental.merge_model_callback.MergeConfig، اختياري) — الإعدادات المستخدمة لعملية الدمج. إذا لم يتم تقديمها، يتم استخدام MergeConfig الافتراضي.
merge_at_every_checkpoint (bool، اختياري، القيمة الافتراضية False) — ما إذا كان سيتم دمج النموذج عند كل نقطة تحقق.
push_to_hub (bool، اختياري، القيمة الافتراضية False) — ما إذا كان سيتم دفع النموذج المدمج إلى Hub بعد الدمج.

TrainerCallback يدمج نموذج السياسة (النموذج قيد التدريب) مع نموذج آخر بناءً على تكوين الدمج.

مثال:

>>> from trl.experimental.merge_model_callback import MergeConfig, MergeModelCallback

>>> config = MergeConfig()
>>> merge_callback = MergeModelCallback(config)
>>> trainer = DPOTrainer(..., callbacks=[merge_callback])

Update on GitHub

←IW-OPD DPO عبر الإنترنت→