pytorch_pfn_extras.nn.modules.lazy_conv#

Classes

pytorch_pfn_extras.nn.modules.lazy_conv.LazyConv1d(...)

Conv1d module with lazy weight initialization.

pytorch_pfn_extras.nn.modules.lazy_conv.LazyConv2d(...)

Conv2d module with lazy weight initialization.

pytorch_pfn_extras.nn.modules.lazy_conv.LazyConv3d(...)

Conv3d module with lazy weight initialization.

pytorch_pfn_extras.nn.modules.lazy_conv.LazyInitializationMixin(...)

A mixin for modules that lazily initialize buffers and parameters.

pytorch_pfn_extras.nn.modules.lazy_conv.UninitializedParameter([...])