| 22 | namespace ppl { namespace nn { namespace python { namespace pmx { |
| 23 | |
| 24 | void RegisterSaveModelOptions(pybind11::module* m) { |
| 25 | pybind11::class_<PySaveModelOptions, PyModelOptionsBase>(*m, "SaveModelOptions") |
| 26 | .def(pybind11::init<>()) |
| 27 | .def_readwrite("external_data_dir", &PySaveModelOptions::external_data_dir); |
| 28 | } |
| 29 | |
| 30 | }}}} // namespace ppl::nn::python::pmx |