Example #1
0
 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()
Example #2
0
 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()