How to add customized models to auto3dseg?

[dirtycomputer]

How to add customized models to auto3dseg?

[Nic-Ma]

Hi @dongyang0122 ,

Could you please help share some guide on this question?

Thanks in advance.

[dirtycomputer]

In this tutorial link
The code below ⬇️ can use dint, swinunetr, segresnet and segresnet2d models.
python -m monai.apps.auto3dseg AutoRunner run --input='./task.yaml'
But I have some customized models, like modifying the segment model architecture. How can I add these customized model to auto3dseg. And I just run python -m monai.apps.auto3dseg AutoRunner run --input='./task.yaml' , so that I can run all models automatically.