Loading issue with pre-trained models

MLearing / Keras-Brats-Improved-Unet3d

3D U-Net: Learning Dense Volumetric Segmentation from Sparse Annotation

MIT License

88 stars 15 forks source link

Loading issue with pre-trained models #6

Open Monish-k opened 3 weeks ago

Monish-k commented 3 weeks ago

Hello,

I am trying to use your pre-trained model as a transfer learning model for inference. Unfortunately, when I try to load them using tensorflow/pytorch/keras I am getting concatenation error.

Please let me know if I am doing anything wrong and what I need to do to debug this issue.

Thank you.