def save_onnx(self, model, path=None): with util.TempAttrChange(self, "path", path): loader = args_util.run_script(self.add_save_onnx, model) return loader()
def save_engine(self, engine, path=None): with util.TempAttrChange(self, "path", path): loader = args_util.run_script(self.add_save_engine, engine) return loader()