Hello,
So I am wanted to implement Attention Unet. But there is some issue coming up while using it.
I am using my own medical dataset and also doing a lot of preprocessing with data.
When I am using your model I get this error.

error1.png1039×310 51.3 KB

#Not able to post more pics due to new user.
#My attention Model is as follows:
#And the Forward loop for the AttUnet is :
#Any ideas why this is happening?