TinyDiff
v0.1.0-3-g301651b
Usage
API
tiny_diff
tiny_diff.constants
tiny_diff.losses
tiny_diff.models
tiny_diff.modules
tiny_diff.modules.attention
tiny_diff.modules.conv
tiny_diff.modules.layer_factory
tiny_diff.modules.nonlinearity
tiny_diff.modules.norm
tiny_diff.modules.resampling
tiny_diff.modules.residual
tiny_diff.schedulers
tiny_diff.utils
TinyDiff
API
tiny_diff
tiny_diff.modules
tiny_diff.modules.norm
View page source
tiny_diff.modules.norm
Classes
GroupNorm
(num_groups, num_channels, *args, ...)
GroupNorm extension.