Hi anand,
People are working hard to resolve this issue at the backend. In the meantime, have a look at this post where some potential solutions are discussed: [Week 3] Image segmentation U-net assignment - #7 by vuqpham
Hope this helps and that you get to pass the assignment.