search

frankkramer-lab / aucmedi

a framework for Automated Classification of Medical Images
https://frankkramer-lab.github.io/aucmedi/
GNU General Public License v3.0
38 stars 12 forks source link

EfficientNet not working #185

Open muellerdo opened 1 year ago

muellerdo commented 1 year ago

Currently, the official implementation of EfficientNetBx in TensorFlow/Keras has currently some issues based on serialization resulting in the lack of saving the model or its weights (which also makes checkpoint training based on val_loss impossible).

https://github.com/keras-team/tf-keras/issues/383

Waiting for fix.

Current solution: downgrade to TensorFlow 2.9, however the ConvNeXt architecture is only functional in TF 2.10.

muellerdo commented 1 year ago

Is fixed on latest keras master branch.

Should be included in tf.keras with the next version.