denoise.model

class denoise.model.unet_bottleneck_gn(*args: Any, **kwargs: Any)[source]

Bases: Module

class denoise.model.unet_box_gn(*args: Any, **kwargs: Any)[source]

Bases: Module

class denoise.model.unet_down(*args: Any, **kwargs: Any)[source]

Bases: Module

class denoise.model.unet_ns_gn(*args: Any, **kwargs: Any)[source]

Bases: Module

class denoise.model.unet_up(*args: Any, **kwargs: Any)[source]

Bases: Module