More Unets

[sr5434]

Hello. I would like to train a third super-resolution Unet to bring the resolution to 2048x2048. How could I go about training that (ideally in a Google Colab instance). Thanks!

[lucidrains]

Super easy, just do it the same way as in the readme and pass in a tuple of 4 unets instead of 3

[lucidrains]

I would also remove all attention from the 4th unet and do random crops

[sr5434]

Thank you @lucidrains. What dataset do you recommend training the Super-Resoluting unets on?

[Feanor007]

@lucidrains what's the reason for disabling all the attention layers in the super-resolution U-Net? Does that help improve the image quality?

[Feanor007]

@lucidrains Thanks for the clear explanation :)