(self)
| 368 | |
| 369 | class BARTEncoderConverter(ModelFileConverter): |
| 370 | def __init__(self): |
| 371 | super().__init__(BARTEncoderTorchFile, BARTEncoderONNXFile, BARTEncoderTRTEngine) |
| 372 | |
| 373 | def torch_to_onnx( |
| 374 | self, output_fpath: str, model: Module, network_metadata: NetworkMetadata |