baby.models.unet

baby.models.unet(generator, flattener, weights={}, **kwargs)