BiaPyX / BiaPy

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

Correct UNETR config, add paths for local upload #34

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. I added section for paths for upload from local machine.