(self, output_dir=default_gen_path)
| 81 | ) |
| 82 | |
| 83 | def convert_to_mge(self, output_dir=default_gen_path): |
| 84 | for model in self.models: |
| 85 | input = "{}/{}.onnx".format(output_dir, model.name) |
| 86 | output = "{}/{}.mge".format(output_dir, model.name) |
| 87 | onnx_to_mge(input, output) |
| 88 | |
| 89 | def make(self, model_dir=""): |
| 90 | if model_dir != "": |