romeokienzler's picture
Add small UNET model (#11)
548ff8a verified