composer.models.unet.unet_hparams#

composer.models.unet.unet_hparams

Functions

dataclass

Returns the same class as was passed in, with dunder methods added based on the fields defined in the class.

Hparams

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

`ModelHparams`	Model Hparams.
`UnetHparams`	composer.models.unet.unet_hparams.UnetHparams

class composer.models.unet.unet_hparams.UnetHparams(initializers: List[composer.models.model_hparams.Initializer] = <factory>, num_classes: Union[int, NoneType] = None)[source]#

Bases: composer.models.model_hparams.ModelHparams

composer.models.unet.unet_hparams.UnetHparams