composer.models.unet.unet_hparams#

YAHP interface for UNet.

Hparams

These classes are used with yahp for YAML-based configuration.

UnetHparams

YAHP interface for UNet.
class composer.models.unet.unet_hparams.UnetHparams(initializers=<factory>, num_classes=None)[source]#

Bases: composer.models.model_hparams.ModelHparams

YAHP interface for UNet.

Parameters

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