Komponen Pipa TFX Pendorong

Komponen Pusher digunakan untuk mendorong model yang divalidasi ke target penerapan selama pelatihan model atau pelatihan ulang. Sebelum penerapan, Pusher mengandalkan satu atau lebih berkah dari komponen validasi lainnya untuk memutuskan apakah akan mendorong model atau tidak.

  • Evaluator memberkati model jika model baru yang dilatih "cukup baik" untuk dimasukkan ke produksi.
  • (Opsional namun disarankan) InfraValidator memberkati model jika model dapat diservis secara mekanis di lingkungan produksi.

Komponen Pusher menggunakan model terlatih dalam format SavedModel , dan menghasilkan SavedModel yang sama, bersama dengan metadata versi.

Menggunakan Komponen Pendorong

Komponen pipeline Pusher biasanya sangat mudah diterapkan dan memerlukan sedikit penyesuaian, karena semua pekerjaan dilakukan oleh komponen Pusher TFX. Kode tipikal terlihat seperti ini:

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

Mendorong model yang dihasilkan dari InfraValidator.

(Dari versi 0.30.0)

InfraValidator juga dapat menghasilkan artefak InfraBlessing yang berisi model dengan pemanasan , dan Pusher dapat mendorongnya seperti artefak 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(...)
)

Detail lebih lanjut tersedia di referensi API Pusher .