kbressem / trainlib

MIT License
4 stars 0 forks source link

make `get_model` more general #58

Closed kbressem closed 1 year ago

kbressem commented 1 year ago

Until now, the default model was monai.networks.nets.UNet. Changing the model required using the patch functionality, which is cumbersome. Now it's possible to define most monai models in the config. Only custom models must still be patched.

I also re-ran make pretty on all files, solving the CI issue.