• Sanjeev Kumar's avatar
    Enable inference config in export step · 90aff5da
    Sanjeev Kumar authored
    Summary:
    - Enable sdk inference config specification in export step. This enables adding the sdk configuration as part of model file in the export step. The sdk config can be specified as infernece_config.yaml and is zipped together with torchscript model. The main goal of sdk configuration is to control the model inference behavior with model.
    - SDK inference config design doc: https://docs.google.com/document/d/1j5qx8IrnFg1DJFzTnu4W8WmXFYJ-AgCDfSQHb2ACJsk/edit
    - One click fblearner pipeline is in next diff on the stack
    
    Differential Revision: D27881742
    
    fbshipit-source-id: 34a3ab7a88f456b74841cf671ea1b3f678cdb733
    90aff5da
exporter.py 5.88 KB