Ryan Thompson created BEAM-14218:
------------------------------------

             Summary: Add resource location hints to base RunInference 
Implementation
                 Key: BEAM-14218
                 URL: https://issues.apache.org/jira/browse/BEAM-14218
             Project: Beam
          Issue Type: Sub-task
          Components: sdk-py-core
            Reporter: Ryan Thompson


Our generic version of RunInference should add resource location hints, similar 
to what TFX-BSL does here.

 

[https://github.com/tensorflow/tfx-bsl/blob/182918aaefd5287e7669d87bfc818155470315aa/tfx_bsl/beam/run_inference.py#L117]

 

There's no reason that these hints shouldn't be just as relevant to sk-learn or 
pytorch large models hosted on GC or wherever these hints are supported.



--
This message was sent by Atlassian Jira
(v8.20.1#820001)

Reply via email to