modelzoo.vision.pytorch.unet.modeling_unet.UNet#

class modelzoo.vision.pytorch.unet.modeling_unet.UNet[source]#

Bases: torch.nn.Module

UNet Models

Methods

`bce_loss`
`forward`
`reset_parameters`

__call__(*args: Any, **kwargs: Any) → Any#: Call self as a function.

__init__(model_params)[source]#

static __new__(cls, *args: Any, **kwargs: Any) → Any#

previous

modelzoo.vision.pytorch.unet.modeling_unet

next

modelzoo.vision.pytorch.unet.run