search

AAitorG / UNETR_2D

2D UNETR implementation in Tensorflow-keras, for EM image segmentation.
MIT License
3 stars 1 forks source link

Error while running locally on windows #1

Closed saurabhsensai closed 6 months ago

saurabhsensai commented 6 months ago

When running on windows locally it is giving error as "The last dimension of the inputs to Dense should be defined. Found None." Its working fine on Google Colab.

The error in file unetr_2d.py on line "encoded_patches = PatchEncoder(num_patches, projection_dim)(patches)"

AAitorG commented 6 months ago

Hi @saurabhsensai,

Thank you for reaching out and reporting this issue. It sounds like the TensorFlow version could indeed be the culprit here. Since it's working on Colab (with TensorFlow v2.15.0), ensuring you're using the same version locally might resolve the issue.

saurabhsensai commented 6 months ago

Great 😃. It actually solved the problem. Thank you very much.

AAitorG commented 6 months ago

You're welcome! I'm glad to hear that the solution worked for you.