Method __init__

demo/HuggingFace/BART/export.py:215–217

(self, model, network_metadata)

class BARTEncoderTRTEngine(TRTEngineFile):

    def __init__(self, model, network_metadata):
        super().__init__(model, BARTEncoderConverter, network_metadata)
        self.max_trt_workspace = 2048

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