pytorch_pfn_extras.nn.modules.lazy_batchnorm

Classes

pytorch_pfn_extras.nn.modules.lazy_batchnorm.LazyBatchNorm1d(...)	BatchNorm1d module with lazy weight initialization.
pytorch_pfn_extras.nn.modules.lazy_batchnorm.LazyBatchNorm2d(...)	BatchNorm2d module with lazy weight initialization.
pytorch_pfn_extras.nn.modules.lazy_batchnorm.LazyBatchNorm3d(...)	BatchNorm3d module with lazy weight initialization.
pytorch_pfn_extras.nn.modules.lazy_batchnorm.LazyInitializationMixin(...)	A mixin for modules that lazily initialize buffers and parameters.
pytorch_pfn_extras.nn.modules.lazy_batchnorm.UninitializedParameter([...])