Dołącz do społeczności SIG TFX-Addons i pomóż ulepszyć TFX! Dołącz do dodatków SIG TFX

Komponent rurociągu Pusher TFX

Komponent Pusher jest używana do pchania ustalonym modelu do wdrożenia docelowego podczas modelu szkolenia lub przekwalifikowania. Przed wdrożeniem Pusher opiera się na co najmniej jednym błogosławieństwie z innych składników walidacji, aby zdecydować, czy wypchnąć model, czy nie.

  • Oceniający błogosławi model nowy wyszkolony model jest „wystarczająco dobre” musi być odłożony do produkcji.
  • (Opcjonalne, ale zalecane) InfraValidator błogosławi model model jest mechanicznie servable w środowisku produkcyjnym.

Składnik A Pusher zużywa wyszkolony model w SavedModel formacie i daje ten sam SavedModel, wraz z wersjami metadanych.

Korzystanie z popychacza

Komponent potoku Pusher jest zazwyczaj bardzo łatwy do wdrożenia i wymaga niewielkiego dostosowania, ponieważ cała praca jest wykonywana przez komponent Pusher TFX. Typowy kod wygląda tak:

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)
  )
)

Popychanie modelu wyprodukowanego z InfraValidatora.

(od wersji 0.30.0)

InfraValidator może również produkować InfraBlessing artefakt zawierający model z rozgrzewki , a popychacz może wypchnąć go podobnie jak Model artefaktu.

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(...)
)

Więcej szczegółów można znaleźć w dokumentacji API Pusher .