modelzoo.vision.pytorch.unet.model.UNetModel

class modelzoo.vision.pytorch.unet.model.UNetModel

Bases: torch.nn.Module

Methods

build_model
forward

__call__(*args: Any, **kwargs: Any) → Any

Call self as a function.

__init__(params)

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

previous

modelzoo.vision.pytorch.unet.model

next

modelzoo.vision.pytorch.unet.modeling_unet