UnetHparams

class composer.models.UnetHparams(initializers=<factory>, num_classes=None)

YAHP interface for UNet.

Parameters

num_classes (int, optional) – The number of classes. Needed for classification tasks. Default: 3.