search

BiaPyX / BiaPy

Open source Python library for building bioimage analysis pipelines
https://BiaPyX.github.io
MIT License
116 stars 28 forks source link

Correct config names and parameters for unetr #33

Closed lenkaback closed 8 months ago

lenkaback commented 8 months ago

I corrected key names for the config file of unetr-based model - VIT_EMBED_DIM and VIT_MLP_RATIO. I changed the default values of VIT_MLP_RATIO to floats, such that you get correct number of the dense layers of the final classifier after multiplying with VIT_EMBED_DIM.