ptrblck:

Could you check the shapes of block2 and block1 and make sure they are equal in all necessary dimensions?


The shapes are as follows.
block1.shape = [1, 16, 124, 124, 124]
block2.shape = [1, 32, 119, 119, 119]
Please, how can I modify the codes so as to make them equal in all necessary dimensions?



 ptrblck:

I’ve implemented a simple UNet a while ago here, which could be a good starter code for your model.


I have also checked this model as suggested. However, I am also facing some troubles with it.
Please, could we first adjust the shapes of block2 and block1? Then, I may open another topic for the model you recommended.
Thank you very much for your time and patience