Hi!
The Kernel in the jupyter notebook keeps dying for me as soon as the U-Net model training starts.
I had one successful run through the training in the past, but I could never repeat that with the same code.
Could it be a bit too much for the notebook? The autograder passes the assignment with full score.