Same problem for me, it looks like Unets don’t like the DataParallel() wrapper too much. People on the forum suggest to parallelize through to_distributed() but as far as I understand this approach is not viable inside a Jupyter Notebook, one has to convert his notebooks to proper python scripts and then run them via command line.