composer.models.unet.unet#
composer.models.unet.unet
Classes
The minimal interface needed to use a model with |
|
The Dice Coefficient for evaluating image segmentation. |
|
|
composer.models.unet.unet.torch.Tensor |
A U-Net model extending |
|
|
composer.models.unet.model.UNet |
Attributes
Any
BatchPair
Metrics
Optional
Tensors
Tuple
log