Bug fixes for VAE regularization and pretrained models.

mist-medical / MIST

MIST: A simple, scalable, and end-to-end framework for 3D medical imaging segmentation.

Apache License 2.0

29 stars 11 forks source link

Closed aecelaya closed 1 month ago

aecelaya commented 1 month ago

Bug fixes for VAE regularization.

We're seeing some strange behavior for pre-trained models. We implemented a fix that seems to work for now, but further investigation is required.

Refactor run.py to production-quality code.

Improved error handling.