Thanks for the great code and model

[kvrooman]

Great work. I've enjoyed the results and experimenting with your concept. Appreciate the effort you made towards sharing your code and pipeline.

I saw a reference to a new and improved Unet version in PyTorch? How goes the experiments with the better version?

[Yuval Nirkin]

It's already much better and more efficient than the old model, but it will only be released with the next paper.