Activations
Module containing helper functions and classes around activations
set_seed
set_seed (seed, deterministic=False)
Hook
Hook (nr, layer, func)
Initialize self. See help(type(self)) for accurate signature.
ActivationStatsS
ActivationStatsS (modules)
Initialize self. See help(type(self)) for accurate signature.
init_weights
init_weights (m)
NormalizationS
NormalizationS (mean, std)
Initialize self. See help(type(self)) for accurate signature.
conv_block
conv_block (in_c, out_c, kernel_size=3, stride=2, act=True, norm=True)
cnn_layers
cnn_layers (act=True)