Method __init__

demo/HuggingFace/BART/export.py:202–204

(self, model, network_metadata)

class BARTDecoderTRTEngine(TRTEngineFile):

    def __init__(self, model, network_metadata):
        super().__init__(model, BARTDecoderConverter, network_metadata)
        self.max_trt_workspace = BARTModelTRTConfig.MAX_DECODER_WORKSPACE_MB[network_metadata.variant]

    def get_network_definition(self, network_definition):
        return add_extra_fp32(network_definition)