Here is the code to the THEANO version which works like a charm!!

  
      
      GitHub
  
  
    

saeedizadi/UNET_THEANO 132

Contribute to saeedizadi/UNET_THEANO development by creating an account on GitHub.


  
  
    
    
  
  


Here’s the PyTorch code not working!
https://github.com/saeedizadi/UNET_POLYP 168