class_docstr: Model class
class_docstr_tok: ["Model","class"]
code: def save_model(self, model_type, output_file):
      if model_type != "onnx":
          return False
      dummy_input = torch.randn(1, 3, 224, 224)
      torch.onnx.export(
          self.model,
          dummy_input,
          output_file,
          input_names=["input0"],
          output_names=["output0"],
      )
      return True
  
file_path: towhee/tests/unittests/serve/triton/test_model_to_triton.py
imports: import unittest
  from pathlib import Path
  from tempfile import TemporaryDirectory
  import torch
  import torchvision
  from towhee.runtime.node_config import NodeConfig
  from towhee.serve.triton.model_to_triton import ModelToTriton
label_desc: Use image node_id 3 for calling the Model obj's underlying member method code with example usage: obj.save_model(model_type, output_file) and returns: True, False