רכיב צינור ה- TFX Pusher

מרכיב Pusher משמש לדחוף מודל תקף עד יעד פריסה במהלך אימון מודל או הכשרה מחדש. לפני הפריסה, Pusher מסתמך על ברכה אחת או יותר ממרכיבי אימות אחרים כדי להחליט אם לדחוף את המודל או לא.

  • למעריכים מברכים את המודל אם המודל המודרך החדש הוא "מספיק טוב" יידחף ייצור.
  • (אופציונאלי אך מומלץ) InfraValidator מברך את המודל אם המודל הוא servable מכאני בסביבת ייצור.

"ממריץ רכיב צורכת מודל הכשרה SavedModel פורמט, ומייצרת אותו SavedModel, יחד עם גרסאות מטה.

שימוש ברכיב ה-Pusher

רכיב צינור של Pusher הוא בדרך כלל קל מאוד לפריסה ודורש מעט התאמה אישית, מכיוון שכל העבודה נעשית על ידי רכיב Pusher TFX. קוד טיפוסי נראה כך:

pusher = Pusher(
  model=trainer.outputs['model'],
  model_blessing=evaluator.outputs['blessing'],
  infra_blessing=infra_validator.outputs['blessing'],
  push_destination=tfx.proto.PushDestination(
    filesystem=tfx.proto.PushDestination.Filesystem(
        base_directory=serving_model_dir)
  )
)

דחיפת דגם שהופק מ-InfraValidator.

(מגרסה 0.30.0)

InfraValidator יכול גם לייצר InfraBlessing חפץ המכיל דגם עם חימום , ו Pusher יכול לדחוף אותו בדיוק כמו Model חפץ.

infra_validator = InfraValidator(
    ...,
    # make_warmup=True will produce a model with warmup requests in its
    # 'blessing' output.
    request_spec=tfx.proto.RequestSpec(..., make_warmup=True)
)

pusher = Pusher(
    # Push model from 'infra_blessing' input.
    infra_blessing=infra_validator.outputs['blessing'],
    push_destination=tfx.proto.PushDestination(...)
)

פרטים נוספים זמינים ב פניית API Pusher .